balthou commited on
Commit
fca4341
·
1 Parent(s): f6b56a2

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
- dl_train = get_dataloader(
79
- {
80
- TRAIN: get_config_dataloader(
81
- audio_root=mixed_audio_root,
82
- mode=TRAIN,
83
- shuffle=True,
84
- batch_size=config[BATCH_SIZE][TRAIN],
85
- augmentation=config[DATALOADER].get(AUGMENTATION, {})
86
- )
87
- },
88
- audio_dataset=RemixedRandomAudioDataset
89
- )[TRAIN]
90
- dl_test = get_dataloader(
91
- {
92
- TEST: get_config_dataloader(
93
- audio_root=mixed_audio_root,
94
- mode=TEST,
95
- shuffle=False,
96
- batch_size=config[BATCH_SIZE][TEST]
97
- )
98
- },
99
- audio_dataset=RemixedFixedAudioDataset
100
- )[TEST]
 
 
 
 
 
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