from local

This commit is contained in:
dohe0342 2023-01-13 16:51:16 +09:00
parent 4d1ace8306
commit 04977d959a
2 changed files with 1 additions and 1 deletions

View File

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