Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix iterator

This commit is contained in:
shahules786 2022-10-26 12:00:57 +05:30
parent f07c8741ba
commit ee40259a8d
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
enhancer/data/dataset.py
View File

@ -302,10 +302,11 @@ class EnhancerDataset(TaskDataset):
if idx >= num_samples: if idx >= num_samples:
idx -= num_samples idx -= num_samples
continue continue
start = 0 else:
if self.duration is not None: start = 0
start = idx * self.stride if self.duration is not None:
return self.prepare_segment(filedict, start) start = idx * self.stride
return self.prepare_segment(filedict, start)
def val__getitem__(self, idx): def val__getitem__(self, idx):
return self.prepare_segment(*self._validation[idx]) return self.prepare_segment(*self._validation[idx])