mattricesound commited on
Commit
6990e4a
·
1 Parent(s): 78ec3fa

Fix render_root naming2

Browse files
Files changed (1) hide show
  1. remfx/datasets.py +3 -3
remfx/datasets.py CHANGED
@@ -20,14 +20,14 @@ class VocalSet(Dataset):
20
  chunk_size_in_sec: int = 3,
21
  effect_types: List[torch.nn.Module] = None,
22
  render_files: bool = True,
23
- rendered_root: str = None,
24
  mode: str = "train",
25
  ):
26
  super().__init__()
27
  self.chunks = []
28
  self.song_idx = []
29
  self.root = Path(root)
30
- self.rendered_root = Path(rendered_root)
31
  self.chunk_size_in_sec = chunk_size_in_sec
32
  self.sample_rate = sample_rate
33
  self.mode = mode
@@ -37,7 +37,7 @@ class VocalSet(Dataset):
37
  self.normalize = effects.LoudnessNormalize(sample_rate, target_lufs_db=-20)
38
  self.effect_types = effect_types
39
 
40
- self.processed_root = self.rendered_root / "processed" / self.mode
41
 
42
  self.num_chunks = 0
43
  print("Total files:", len(self.files))
 
20
  chunk_size_in_sec: int = 3,
21
  effect_types: List[torch.nn.Module] = None,
22
  render_files: bool = True,
23
+ render_root: str = None,
24
  mode: str = "train",
25
  ):
26
  super().__init__()
27
  self.chunks = []
28
  self.song_idx = []
29
  self.root = Path(root)
30
+ self.render_root = Path(render_root)
31
  self.chunk_size_in_sec = chunk_size_in_sec
32
  self.sample_rate = sample_rate
33
  self.mode = mode
 
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))