From 40d26d39e1a5545a42782a59daf62613a3a95972 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 23 Mar 2023 23:05:15 +0900
Subject: [PATCH] from local

---
 .../.train_uda.py.swp                         | Bin 98304 -> 98304 bytes
 .../train_uda.py                              |   7 +++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

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 6f65548e26ee43c0cc2ca4f7571edf65129d8789..c52e5627f3ccbd6f695583d94fed9fbce1fb7870 100644
GIT binary patch
delta 361
zcmZo@U~6b#6HPJ*^Ym4))H7fJ0s#gFpBR~x_Zvl}1sNlo4F$Iw3Nl`f<K<vtVCZ0m
zD2SXcSipF^-hhjNK@f=FaWXKR2jaOv90|n6Kr9Eue>fNzz60@UAT9>tA|Or$VkIE{
z#17O0#Or`~G7uL5u@4Y`W@BJ@2*lfgcpDJM1F<X+%K$M05N}{*U|0{t39Jmt43a>`
zdlsMrfOtL-CjhZL5Z?p3b2AWkFmJrb$F%)jF{3aWqt$fADn{c}1t=&fO3ciQPsvfR
zRe&+04PqJ4rQ=Ie5|Nb{LX>2tDU?(eq-wylYAV>;D&%Aqmso*K*uIOMv7U8$LO3JG
J^xgH0t^kL!P1FDY

delta 249
zcmXBKKTE?<00i)RiTR@v@PScEQ|Kos3ARJ`iga+(sGy~U(phlQB@a`fW1&M3`X$mO
zP=`*9gHv^}Um|<A9`xV`2X{)DN||othu!OXyWxX50h*8Apvc^+6k!s`JdonA=e%h^
zsqYVADxbx7+&ls_S{?$QG@^)}NYcgz{^^ItbWbfhrLDz{=JY^UR3nd8CXmoO4e5+r
zDhwc|Te>lD4uhpQ06wTg6)N>?t+c80Yv+>}acriuS8<UqED;-N|4!KHS1c}fX4@?*

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 a729090f9..c25ab54aa 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
@@ -1016,9 +1016,12 @@ def train_one_epoch(
     if params.multi_optim:
         optimizer_enc, optimizer_dec = optimizer[0], optimizer[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(zip(train_dl, train_dl_uda))
+    for batch_idx, batch in enumerate(zip(train_dl, train_dl_uda)):
         if batch_idx < cur_batch_idx:
             continue
         cur_batch_idx = batch_idx