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
d3a85874f8
commit
7210ba025b
Binary file not shown.
@ -1311,7 +1311,7 @@ def run(rank, world_size, args, wb=None):
|
||||
and ("scheduler" in checkpoints or "scheduler_enc" in checkpoints)
|
||||
and checkpoints["scheduler"] is not None
|
||||
):
|
||||
if params.multi_optim:
|
||||
if params.multi_optim and "scheduler_enc" in checkpoints:
|
||||
scheduler_enc.load_state_dict(checkpoints["scheduler_enc"])
|
||||
scheduler_dec.load_state_dict(checkpoints["scheduler_dec"])
|
||||
else:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user