from local

This commit is contained in:
dohe0342 2023-02-20 10:32:18 +09:00
parent 06b81a042d
commit c6236f455f
2 changed files with 1 additions and 1 deletions

View File

@ -1064,7 +1064,7 @@ def run(rank, world_size, args):
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")
scheduler.load_state_dict(checkpoints["scheduler"]) #scheduler.load_state_dict(checkpoints["scheduler"])
if params.print_diagnostics: if params.print_diagnostics:
opts = diagnostics.TensorDiagnosticOptions( opts = diagnostics.TensorDiagnosticOptions(