mattricesound commited on
Commit
57c446b
·
1 Parent(s): b86a99b

Fix render_path location so it doesn't overlap on same deivce

Browse files
Files changed (2) hide show
  1. remfx/datasets.py +5 -3
  2. scripts/test.py +0 -1
remfx/datasets.py CHANGED
@@ -36,9 +36,11 @@ class VocalSet(Dataset):
36
  self.files = sorted(list(mode_path.glob("./**/*.wav")))
37
  self.normalize = effects.LoudnessNormalize(sample_rate, target_lufs_db=-20)
38
  self.effect_types = effect_types
39
-
40
- self.processed_root = self.render_root / "processed" / self.mode
41
-
 
 
42
  self.num_chunks = 0
43
  print("Total files:", len(self.files))
44
  print("Processing files...")
 
36
  self.files = sorted(list(mode_path.glob("./**/*.wav")))
37
  self.normalize = effects.LoudnessNormalize(sample_rate, target_lufs_db=-20)
38
  self.effect_types = effect_types
39
+ effect_str = "_".join([e.__class__.__name__ for e in self.effect_types])
40
+ self.processed_root = self.render_root / "processed" / effect_str / self.mode
41
+ if self.processed_root.exists():
42
+ print("Found processed files.")
43
+ render_files = False
44
  self.num_chunks = 0
45
  print("Total files:", len(self.files))
46
  print("Processing files...")
scripts/test.py CHANGED
@@ -14,7 +14,6 @@ def main(cfg: DictConfig):
14
  # Apply seed for reproducibility
15
  if cfg.seed:
16
  pl.seed_everything(cfg.seed)
17
- cfg.render_files = False
18
  log.info(f"Instantiating datamodule <{cfg.datamodule._target_}>.")
19
  datamodule = hydra.utils.instantiate(cfg.datamodule, _convert_="partial")
20
  log.info(f"Instantiating model <{cfg.model._target_}>.")
 
14
  # Apply seed for reproducibility
15
  if cfg.seed:
16
  pl.seed_everything(cfg.seed)
 
17
  log.info(f"Instantiating datamodule <{cfg.datamodule._target_}>.")
18
  datamodule = hydra.utils.instantiate(cfg.datamodule, _convert_="partial")
19
  log.info(f"Instantiating model <{cfg.model._target_}>.")