From ee40259a8df9cf4467a14b7b16ba9ee742c660cd Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 26 Oct 2022 12:00:57 +0530 Subject: [PATCH] fix iterator --- enhancer/data/dataset.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index 34ecb8f..f05fd6b 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -302,10 +302,11 @@ class EnhancerDataset(TaskDataset): if idx >= num_samples: idx -= num_samples continue - start = 0 - if self.duration is not None: - start = idx * self.stride - return self.prepare_segment(filedict, start) + else: + start = 0 + if self.duration is not None: + start = idx * self.stride + return self.prepare_segment(filedict, start) def val__getitem__(self, idx): return self.prepare_segment(*self._validation[idx])