mattricesound commited on
Commit
83de436
·
1 Parent(s): 3163835

Fix input, target getting flipped

Browse files
Files changed (1) hide show
  1. 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
- x, y, 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", x, self.sample_rate)
93
- torchaudio.save(output_dir / "target.wav", y, self.sample_rate)
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: