from local

This commit is contained in:
dohe0342 2023-02-03 01:24:01 +09:00
parent af7c1f3162
commit 93e4e63050
2 changed files with 1 additions and 1 deletions

View File

@ -720,7 +720,7 @@ def main():
if world_size > 1:
mp.spawn(run, args=(world_size, args, wb), nprocs=world_size, join=True)
else:
run(rank=0, world_size=1, args=args)
run(rank=0, world_size=1, args=args, wb=wb)
torch.set_num_threads(1)