Spaces:
Sleeping
Sleeping
Commit
·
83de436
1
Parent(s):
3163835
Fix input, target getting flipped
Browse files- remfx/datasets.py +3 -3
remfx/datasets.py
CHANGED
@@ -86,11 +86,11 @@ class VocalSet(Dataset):
|
|
86 |
# Skip if chunk is too small
|
87 |
continue
|
88 |
|
89 |
-
|
90 |
output_dir = self.proc_root / str(self.num_chunks)
|
91 |
output_dir.mkdir(exist_ok=True)
|
92 |
-
torchaudio.save(output_dir / "input.wav",
|
93 |
-
torchaudio.save(output_dir / "target.wav",
|
94 |
torch.save(effect, output_dir / "effect.pt")
|
95 |
self.num_chunks += 1
|
96 |
else:
|
|
|
86 |
# Skip if chunk is too small
|
87 |
continue
|
88 |
|
89 |
+
dry, wet, effect = self.process_effects(resampled_chunk)
|
90 |
output_dir = self.proc_root / str(self.num_chunks)
|
91 |
output_dir.mkdir(exist_ok=True)
|
92 |
+
torchaudio.save(output_dir / "input.wav", wet, self.sample_rate)
|
93 |
+
torchaudio.save(output_dir / "target.wav", dry, self.sample_rate)
|
94 |
torch.save(effect, output_dir / "effect.pt")
|
95 |
self.num_chunks += 1
|
96 |
else:
|