From 9c7a650130aa9ba896846eb35ff05942ecda1395 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Fri, 21 Oct 2022 11:37:26 +0530
Subject: [PATCH] div by batchsize in __len__

---
 enhancer/data/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py
index 9b512ab..cc89083 100644
--- a/enhancer/data/dataset.py
+++ b/enhancer/data/dataset.py
@@ -351,7 +351,7 @@ class EnhancerDataset(TaskDataset):
 
     def train__len__(self):
 
-        return sum([len(item) for item in self.train_data])
+        return sum([len(item) for item in self.train_data]) // self.batch_size
 
     def val__len__(self):
         return len(self._validation)