diff --git a/trainer/dataset.py b/trainer/dataset.py index a6485eea3a9..49332b55add 100644 --- a/trainer/dataset.py +++ b/trainer/dataset.py @@ -150,7 +150,7 @@ def __init__(self, root, opt): self.nSamples = len(self.df) if self.opt.data_filtering_off: - self.filtered_index_list = [index + 1 for index in range(self.nSamples)] + self.filtered_index_list = [index for index in range(self.nSamples)] else: self.filtered_index_list = [] for index in range(self.nSamples):