fix printing, gpu order

Author: victor-shepardson

notochord/notochord/data.py

@@ -55,9 +55,9 @@ def _remap_anonymous_instruments(self, program: torch.Tensor) -> torch.Tensor:
         new_program = program.clone()
 
         if len(anon_melodic)>self.n_anon:
-            print(f'warning: {anon_melodic} > {self.n_anon} anon melodic instruments')
+            print(f'warning: {len(anon_melodic)} > {self.n_anon} anon melodic instruments')
         if len(anon_drum)>self.n_anon:
-            print(f'warning: {anon_drum} > {self.n_anon} anon drum instruments')
+            print(f'warning: {len(anon_drum)} > {self.n_anon} anon drum instruments')
 
         i = 0
         for pr in anon_melodic:

notochord/notochord/train.py

@@ -1,3 +1,7 @@
+import os
+# fix torch device order to be same as nvidia-smi order
+os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
+
 from pathlib import Path
 import random
 from collections import defaultdict