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

---
 .../.train_uda.py.swp                         | Bin 94208 -> 98304 bytes
 .../train_uda.py                              |   2 +-
 2 files changed, 1 insertion(+), 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 54db45acb814b8d247c9cba5e36ca75cce9cd933..b1f5bbaddcfc883a133804e920bfb276c9fa58f5 100644
GIT binary patch
delta 876
zcmXxhUr19?9KiA4ahvXL-OW3&6--<CU`oUoRtT;h>Sg{5qFK5I`6na_f(WA6*xo7%
zjjmErK2*{}L}@;_vLGC&h0rDnk(865=plL#)>G8?79IF-&%KBH=iG0m=*VQst9nD7
z(e_}|&X6!FMH-I<`e!GqzI)ZiNn53^l+QE_zu~u*8L@u{S(w~x%z2&#?M2<8i=2dY
zw{OHyt!eA53Hz}L-!_Qk@d{7y04XG~7lIYH$Oq&xgJE3406G!IR;;;13K+vMPT&Y?
z;lq+9X~}6KgGgdG8nEgVd5am0<0*zPh&Z~j55Fozrttz-(1#A}Kpj@f*#Pci1c$H<
zza1i<kaHwhWRgTJ41BeVe8CXf(S{I;Ha-m35yckN;9VKla1b7RRw7xvS7L@z{;3~I
zopPIXZS@twGiQVGb3tqF_&YaWt_RHN^?*uD<#iS6P51P(X`T`0eO9CY375ST!434G
z1$(d^)%fWVDItrCh~Y3=&<qb0)+n6EW86U;t*C)Vr^YII3mC;UbRYsJN+cb}7`ku(
z&1gV9G&r$rrNwic$1XI&hL6;p$3xsh4Alt0tDP2EAnRRRz)?gHK#82AxQZSehlV+F
u&f*ce(FGTBl)Q~wh{A<8{ME}yAZ(u~%$J$^qVwkK0cAHMDVJ3$>;40p6Nrrf

delta 461
zcmXBQJ4nKD5Ww;K`F`{-9i$~f!KJ07CYOqmFia?d8XO7{grEl$6eQ>n(Heik$)Krd
zQ4_PV=iFl|dLZN$K^83~)Hgc#z;SmR++{MBOx5g(2PTH2-e6}y2$x7T)|MD5dR|+U
zr(|?#N+or-yR{T{A2n}Qy$4=XRr4`<r~b#2TW%Fe?`k3`BoKrH^(K)gT;d$t$YBaT
zG@K&esN)%B>|q1T7{>q%cz1|gpp12_AcP)#+9fSDyGR~6Od<loi;d;D#uW<4BZXB=
zqa6=ck$dc84fBX1jE02+aE3!HVh~>Zm_=&l7)_EKLg<DKRuoJkbC|^#6a+=1$PD_?
xhc^Q?ET9V}{QdY=-#PK?+KnxHoK&Vt?~*!oW`mmQ&__os`s0Vc@}((D{1+S`Q(gc7

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 0dc10ae11..11ae126b7 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
@@ -967,7 +967,7 @@ def train_one_epoch(
     optimizer: torch.optim.Optimizer or [torch.optim.Optimizer, torch.optim.Optimizer],
     scheduler: LRSchedulerType or [LRSchedulerType, LRSchedulerType],
     sp: spm.SentencePieceProcessor,
-    train_dl: torch.utils.data.DataLoader,
+    train_dl: torch.utils.data.DataLoader or [torch.utils.data.DataLoader, torch.utils.data.DataLoader],
     valid_dl: torch.utils.data.DataLoader,
     scaler: GradScaler,
     model_avg: Optional[nn.Module] = None,