From 6314d210c35f89fe61698d13213a613270628d3e Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 22 Oct 2022 11:05:19 +0530 Subject: [PATCH] debug git commit -m debug ' --- enhancer/data/dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index 05dd287..c808c8e 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -360,7 +360,9 @@ class EnhancerDataset(TaskDataset): train_data = self.train_data else: train_data = worker_info.dataset.data - return sum([len(item) for item in train_data]) // (self.batch_size) + len = sum([len(item) for item in train_data]) // (self.batch_size) + print(len) + return len def val__len__(self): return len(self._validation)