from local

This commit is contained in:
dohe0342 2023-02-03 01:25:51 +09:00
parent 223558f0c1
commit 3fca4c4221
2 changed files with 1 additions and 0 deletions

View File

@ -691,6 +691,7 @@ def run(rank, world_size, args, wb=None):
valid_dl=valid_dl, valid_dl=valid_dl,
tb_writer=tb_writer, tb_writer=tb_writer,
world_size=world_size, world_size=world_size,
wb=wb,
) )
save_checkpoint( save_checkpoint(