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:05:15 +09:00
parent 6179041fc8
commit 40d26d39e1
2 changed files with 5 additions and 2 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.

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

@ -1017,8 +1017,11 @@ 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]
if type(train_dl) == list:
train_dl_uda = train_dl[1]
train_dl = train_dl[0]
#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)) 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