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)