from local

This commit is contained in:
dohe0342 2023-03-29 22:06:05 +09:00
parent 8efc363819
commit 513776a3a9
2 changed files with 2 additions and 0 deletions

View File

@ -1467,6 +1467,7 @@ def run(rank, world_size, args, wb=None):
diagnostic.print_diagnostics()
break
'''
if epoch % 50 == 0:
save_checkpoint(
params=params,
@ -1478,6 +1479,7 @@ def run(rank, world_size, args, wb=None):
scaler=scaler,
rank=rank,
)
'''
logging.info("Done!")