update reqs

requirements.txt

@@ -1,3 +1,4 @@
 batch_processing
 interactive-pipe>=0.7.0
 torch>=2.0.0
+torchaudio

src/gyraudio/audio_separation/experiment_tracking/experiments.py

@@ -75,29 +75,34 @@ def get_experience(exp_major: int, exp_minor: int = 0, snr_filter_test=None, dry
                 AUG_TRIM: {LENGTHS: [None, None], LENGTH_DIVIDER: config[DATALOADER][AUGMENTATION]
                            [AUG_TRIM][LENGTH_DIVIDER], TRIM_PROB: -1.}
             }
-        dl_train = get_dataloader(
-            {
-                TRAIN: get_config_dataloader(
-                    audio_root=mixed_audio_root,
-                    mode=TRAIN,
-                    shuffle=True,
-                    batch_size=config[BATCH_SIZE][TRAIN],
-                    augmentation=config[DATALOADER].get(AUGMENTATION, {})
-                )
-            },
-            audio_dataset=RemixedRandomAudioDataset
-        )[TRAIN]
-        dl_test = get_dataloader(
-            {
-                TEST: get_config_dataloader(
-                    audio_root=mixed_audio_root,
-                    mode=TEST,
-                    shuffle=False,
-                    batch_size=config[BATCH_SIZE][TEST]
-                )
-            },
-            audio_dataset=RemixedFixedAudioDataset
-        )[TEST]
+        try:
+            dl_train = get_dataloader(
+                {
+                    TRAIN: get_config_dataloader(
+                        audio_root=mixed_audio_root,
+                        mode=TRAIN,
+                        shuffle=True,
+                        batch_size=config[BATCH_SIZE][TRAIN],
+                        augmentation=config[DATALOADER].get(AUGMENTATION, {})
+                    )
+                },
+                audio_dataset=RemixedRandomAudioDataset
+            )[TRAIN]
+            dl_test = get_dataloader(
+                {
+                    TEST: get_config_dataloader(
+                        audio_root=mixed_audio_root,
+                        mode=TEST,
+                        shuffle=False,
+                        batch_size=config[BATCH_SIZE][TEST]
+                    )
+                },
+                audio_dataset=RemixedFixedAudioDataset
+            )[TEST]
+        except Exception as e:
+            dl_train = None
+            dl_test = None
+            pass
         dataloaders = {
             TRAIN: dl_train,
             TEST: dl_test

src/gyraudio/audio_separation/visualization/interactive_audio.py

@@ -276,7 +276,7 @@ def parse_command_line_gradio(parser: Batch = None, gradio_demo=True) -> argpars
         parser = parse_command_line_audio_load()
     default_device = "cuda" if torch.cuda.is_available() else "cpu"
     iparse = parser.add_argument_group("Audio separation visualization")
-    iparse.add_argument("-e",  "--experiments", type=int, nargs="+", default=3001,
+    iparse.add_argument("-e",  "--experiments", type=int, nargs="+", default=[3001,],
                         help="Experiment ids to be inferred sequentially")
     iparse.add_argument("-p", "--interactive", default=True,
                         action="store_true", help="Play = Interactive mode")
@@ -286,7 +286,6 @@ def parse_command_line_gradio(parser: Batch = None, gradio_demo=True) -> argpars
                         choices=["cpu", "cuda"] if default_device == "cuda" else ["cpu"])
     iparse.add_argument("-gui", "--gui", type=str,
                         default="gradio", choices=["qt", "mpl", "gradio"])
-    pri
     return parser