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

add arg

This commit is contained in:
shahules786 2022-10-20 21:03:38 +05:30
parent f2561d7cf7
commit ba10719520
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

@ -185,7 +185,7 @@ class TaskDataset(pl.LightningDataModule):
output["noisy"] = torch.stack(output["noisy"], dim=0) output["noisy"] = torch.stack(output["noisy"], dim=0)
return output return output
def worker_init_fn(self): def worker_init_fn(self, _):
worker_info = torch.utils.data.get_worker_info() worker_info = torch.utils.data.get_worker_info()
dataset = worker_info.dataset dataset = worker_info.dataset
worker_id = worker_info.id worker_id = worker_info.id