This commit is contained in:
shahules786 2022-09-23 18:22:25 +05:30
parent 10eaada080
commit 06afc95701
1 changed files with 1 additions and 1 deletions

View File

@ -18,7 +18,7 @@ def create_unique_rng(epoch:int):
worker_info = torch.utils.data.get_worker_info()
if worker_info is not None:
num_workers = worker_info.num_workers
worker_id = worker_info.worker_id
worker_id = worker_info.id
else:
num_workers = 1
worker_id = 0