mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-03-23 23:03:57 +09:00
parent 0d21e11062
commit 6179041fc8
2 changed files with 2 additions and 1 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_uda.py.swp
View File

Binary file not shown.

3
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_uda.py
View File

@ -1017,7 +1017,8 @@ def train_one_epoch(
optimizer_enc, optimizer_dec = optimizer[0], optimizer[1] optimizer_enc, optimizer_dec = optimizer[0], optimizer[1]
scheduler_enc, scheduler_dec = scheduler[0], scheduler[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):
for batch_idx, batch in enumerate(zip(train_dl, train_dl_uda))
if batch_idx < cur_batch_idx: if batch_idx < cur_batch_idx:
continue continue
cur_batch_idx = batch_idx cur_batch_idx = batch_idx