Spaces:
Running
Running
update reqs
Browse files
requirements.txt
CHANGED
@@ -1,3 +1,4 @@
|
|
1 |
batch_processing
|
2 |
interactive-pipe>=0.7.0
|
3 |
torch>=2.0.0
|
|
|
|
1 |
batch_processing
|
2 |
interactive-pipe>=0.7.0
|
3 |
torch>=2.0.0
|
4 |
+
torchaudio
|
src/gyraudio/audio_separation/experiment_tracking/experiments.py
CHANGED
@@ -75,29 +75,34 @@ def get_experience(exp_major: int, exp_minor: int = 0, snr_filter_test=None, dry
|
|
75 |
AUG_TRIM: {LENGTHS: [None, None], LENGTH_DIVIDER: config[DATALOADER][AUGMENTATION]
|
76 |
[AUG_TRIM][LENGTH_DIVIDER], TRIM_PROB: -1.}
|
77 |
}
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
|
|
|
|
|
|
|
|
|
|
101 |
dataloaders = {
|
102 |
TRAIN: dl_train,
|
103 |
TEST: dl_test
|
|
|
75 |
AUG_TRIM: {LENGTHS: [None, None], LENGTH_DIVIDER: config[DATALOADER][AUGMENTATION]
|
76 |
[AUG_TRIM][LENGTH_DIVIDER], TRIM_PROB: -1.}
|
77 |
}
|
78 |
+
try:
|
79 |
+
dl_train = get_dataloader(
|
80 |
+
{
|
81 |
+
TRAIN: get_config_dataloader(
|
82 |
+
audio_root=mixed_audio_root,
|
83 |
+
mode=TRAIN,
|
84 |
+
shuffle=True,
|
85 |
+
batch_size=config[BATCH_SIZE][TRAIN],
|
86 |
+
augmentation=config[DATALOADER].get(AUGMENTATION, {})
|
87 |
+
)
|
88 |
+
},
|
89 |
+
audio_dataset=RemixedRandomAudioDataset
|
90 |
+
)[TRAIN]
|
91 |
+
dl_test = get_dataloader(
|
92 |
+
{
|
93 |
+
TEST: get_config_dataloader(
|
94 |
+
audio_root=mixed_audio_root,
|
95 |
+
mode=TEST,
|
96 |
+
shuffle=False,
|
97 |
+
batch_size=config[BATCH_SIZE][TEST]
|
98 |
+
)
|
99 |
+
},
|
100 |
+
audio_dataset=RemixedFixedAudioDataset
|
101 |
+
)[TEST]
|
102 |
+
except Exception as e:
|
103 |
+
dl_train = None
|
104 |
+
dl_test = None
|
105 |
+
pass
|
106 |
dataloaders = {
|
107 |
TRAIN: dl_train,
|
108 |
TEST: dl_test
|
src/gyraudio/audio_separation/visualization/interactive_audio.py
CHANGED
@@ -276,7 +276,7 @@ def parse_command_line_gradio(parser: Batch = None, gradio_demo=True) -> argpars
|
|
276 |
parser = parse_command_line_audio_load()
|
277 |
default_device = "cuda" if torch.cuda.is_available() else "cpu"
|
278 |
iparse = parser.add_argument_group("Audio separation visualization")
|
279 |
-
iparse.add_argument("-e", "--experiments", type=int, nargs="+", default=3001,
|
280 |
help="Experiment ids to be inferred sequentially")
|
281 |
iparse.add_argument("-p", "--interactive", default=True,
|
282 |
action="store_true", help="Play = Interactive mode")
|
@@ -286,7 +286,6 @@ def parse_command_line_gradio(parser: Batch = None, gradio_demo=True) -> argpars
|
|
286 |
choices=["cpu", "cuda"] if default_device == "cuda" else ["cpu"])
|
287 |
iparse.add_argument("-gui", "--gui", type=str,
|
288 |
default="gradio", choices=["qt", "mpl", "gradio"])
|
289 |
-
pri
|
290 |
return parser
|
291 |
|
292 |
|
|
|
276 |
parser = parse_command_line_audio_load()
|
277 |
default_device = "cuda" if torch.cuda.is_available() else "cpu"
|
278 |
iparse = parser.add_argument_group("Audio separation visualization")
|
279 |
+
iparse.add_argument("-e", "--experiments", type=int, nargs="+", default=[3001,],
|
280 |
help="Experiment ids to be inferred sequentially")
|
281 |
iparse.add_argument("-p", "--interactive", default=True,
|
282 |
action="store_true", help="Play = Interactive mode")
|
|
|
286 |
choices=["cpu", "cuda"] if default_device == "cuda" else ["cpu"])
|
287 |
iparse.add_argument("-gui", "--gui", type=str,
|
288 |
default="gradio", choices=["qt", "mpl", "gradio"])
|
|
|
289 |
return parser
|
290 |
|
291 |
|