from local

This commit is contained in:
dohe0342 2023-02-20 10:19:33 +09:00
parent ff53bc629f
commit 06b81a042d
3 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

@ -1060,7 +1060,7 @@ def run(rank, world_size, args):
if checkpoints and checkpoints.get("optimizer") is not None: if checkpoints and checkpoints.get("optimizer") is not None:
logging.info("Loading optimizer state dict") logging.info("Loading optimizer state dict")
optimizer.load_state_dict(checkpoints["optimizer"]) #optimizer.load_state_dict(checkpoints["optimizer"])
if checkpoints and checkpoints.get("scheduler") is not None: if checkpoints and checkpoints.get("scheduler") is not None:
logging.info("Loading scheduler state dict") logging.info("Loading scheduler state dict")