fix typo
This commit is contained in:
parent
10eaada080
commit
06afc95701
|
|
@ -18,7 +18,7 @@ def create_unique_rng(epoch:int):
|
||||||
worker_info = torch.utils.data.get_worker_info()
|
worker_info = torch.utils.data.get_worker_info()
|
||||||
if worker_info is not None:
|
if worker_info is not None:
|
||||||
num_workers = worker_info.num_workers
|
num_workers = worker_info.num_workers
|
||||||
worker_id = worker_info.worker_id
|
worker_id = worker_info.id
|
||||||
else:
|
else:
|
||||||
num_workers = 1
|
num_workers = 1
|
||||||
worker_id = 0
|
worker_id = 0
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue