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
088a522667
commit
ca612e1c01
Binary file not shown.
@ -926,6 +926,10 @@ def train_one_epoch(
|
||||
|
||||
cur_batch_idx = params.get("cur_batch_idx", 0)
|
||||
|
||||
if params.multi_optim:
|
||||
optimizer_enc, optimizer_dec = optimizer[0], optimizer[1]
|
||||
scheduler_enc, scheduler_dec = scheduler[0], scheduler[1]
|
||||
|
||||
for batch_idx, batch in enumerate(train_dl):
|
||||
if batch_idx < cur_batch_idx:
|
||||
continue
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user