From 6179041fc8c5690fb39f27bd28ce6aa2658c4ab5 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 23 Mar 2023 23:03:57 +0900
Subject: [PATCH] from local

---
 .../.train_uda.py.swp                         | Bin 98304 -> 98304 bytes
 .../train_uda.py                              |   3 ++-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_uda.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_uda.py.swp
index b1f5bbaddcfc883a133804e920bfb276c9fa58f5..6f65548e26ee43c0cc2ca4f7571edf65129d8789 100644
GIT binary patch
delta 257
zcmZo@U~6b#+h8Eb7}jhkxZO~Y@p2q*6%zx)J7xw31_1_!u<3#YjHm0FxEL7rb22c@
z1mYGTP6FasAQlATn;Z-bH-Pvw5O)A^1Q447u?P_VWCzLt@g^XS1Y&m}mIY!NAU?{*
zz|aiDO+cIm#9}}!3dF})85rt;I24FOfS8q)L73qf3j;$d5W4`eBoM!2-gvQrX)^C6
zj_C)|8Cka5l`y_$RjSG?&?qTN%*=~V$<a}OvEoZp5;Zj$6o5c^doMfV8rJCv;fx&9
Jch@tz0sxXSG(G?T

delta 209
zcmWO0y$-=p00z+SR&W0jsk_%ZXm_K*12ldT!W&3TrryLNJU~epECvgv4mJ`hEMoQw
zMpoZ&_M^3>wY9G(5syQag-X1-#ghR<CV&zUr)nVXNx=i&E|AlP))Z5j0`lqM081KC
zld2TZ*9KNJqdxV>rK<&8XiIHskw<3<END)3@+5rR_?b|LBJ=RfjqEcNKbh@pF_mer
KA)NG>h}s|GD<%j4

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_uda.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_uda.py
index 11ae126b7..a729090f9 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_uda.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_uda.py
@@ -1017,7 +1017,8 @@ def train_one_epoch(
         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):
+    for batch_idx, batch in enumerate(zip(train_dl, train_dl_uda))
         if batch_idx < cur_batch_idx:
             continue
         cur_batch_idx = batch_idx