Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

div by batchsize in __len__

This commit is contained in:
shahules786 2022-10-21 11:37:26 +05:30
parent 5d7ea582c9
commit 9c7a650130
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/data/dataset.py
View File

@ -351,7 +351,7 @@ class EnhancerDataset(TaskDataset):
def train__len__(self): 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): def val__len__(self):
return len(self._validation) return len(self._validation)