mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-10 13:56:19 +09:00
parent 088a522667
commit ca612e1c01
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
View File

Binary file not shown.

4
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -926,6 +926,10 @@ def train_one_epoch(
cur_batch_idx = params.get("cur_batch_idx", 0) 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): for batch_idx, batch in enumerate(train_dl):
if batch_idx < cur_batch_idx: if batch_idx < cur_batch_idx:
continue continue