From a6b0970f7448d4eb0be29fcee7bb4cb0ed2ed6fa Mon Sep 17 00:00:00 2001 From: ShagunDwivedi <711shagun@gmail.com> Date: Wed, 29 Jan 2025 12:24:50 +0530 Subject: [PATCH] To Fix KeyError when data_filtering_off set to True --- trainer/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):