Spaces:
Sleeping
Sleeping
Commit
·
57c446b
1
Parent(s):
b86a99b
Fix render_path location so it doesn't overlap on same deivce
Browse files- remfx/datasets.py +5 -3
- 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_}>.")
|