from local

This commit is contained in:
dohe0342 2023-04-11 14:05:12 +09:00
parent 717e0b29f3
commit 6033205ae7
2 changed files with 1 additions and 1 deletions

View File

@ -1364,7 +1364,7 @@ def run(rank, world_size, args, wb=None):
scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs)
if checkpoints and ("optimizer" in checkpoints or "optimizer_enc" in checkpoints):
if params.multi_optim or 1:
if params.multi_optim:
logging.info("Loading optimizer state dict")
optimizer_enc.load_state_dict(checkpoints["optimizer_enc"])
optimizer_dec.load_state_dict(checkpoints["optimizer_dec"])