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

debug

This commit is contained in:
shahules786 2022-10-21 17:17:02 +05:30
parent 20c12556ff
commit a7fb27bb0f
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
enhancer/data/dataset.py
View File

@ -353,8 +353,11 @@ class EnhancerDataset(TaskDataset):
} }
def train__len__(self): def train__len__(self):
worker_info = torch.utils.data.get_worker_info()
return sum([len(item) for item in self.train_data]) // self.batch_size num_workers = worker_info.num_workers if worker_info else 1
return sum([len(item) for item in self.train_data]) // (
self.batch_size * num_workers
)
def val__len__(self): def val__len__(self):
return len(self._validation) return len(self._validation)