mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
717e0b29f3
commit
6033205ae7
Binary file not shown.
@ -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"])
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user