from local

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

View File

@ -1648,6 +1648,7 @@ def run_adapter(rank, world_size, args, wb=None):
diagnostic.print_diagnostics()
break
'''
if epoch % 10 == 0:
save_checkpoint(
params=params,
@ -1659,6 +1660,7 @@ def run_adapter(rank, world_size, args, wb=None):
scaler=scaler,
rank=rank,
)
'''
logging.info("Done!")