fix iterator

2022-10-26 12:00:57 +05:30 · 2022-10-26 12:00:57 +05:30 · ee40259a8d
parent f07c8741ba
commit ee40259a8d
1 changed files with 5 additions and 4 deletions
--- a/enhancer/data/dataset.py
+++ b/enhancer/data/dataset.py
@ -302,6 +302,7 @@ class EnhancerDataset(TaskDataset):
            if idx >= num_samples:
                idx -= num_samples
                continue
+            else:
                start = 0
                if self.duration is not None:
                    start = idx * self.stride