From ca612e1c012b410e511b2b263daae93026f30ceb Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 13:56:19 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 94208 -> 98304 bytes
 .../train.py                                  |   4 ++++
 2 files changed, 4 insertions(+)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index 869c864c5631a1de7d277de1cbf85bdc8acd01f5..08b51920a580ff8a0112da7f49763e7387270c9f 100644
GIT binary patch
delta 858
zcmYk)T}V@59LMqhX|LAY9L^~eb*2SDTdWr^7DchH7KueBgbbV8Vu$&1rZ&x+4XTT}
zaKwZmL=kuq2tf^!Zp^Zxi!>?<B?2KYyhuU1iRkTnYE%z=c>d3M&L7Tso?m{+lFwVc
zUBUK9iyGP;6h@6mn|my-e7@`b<Wjt4Q_Y;B_=~UH_3nCOpAq>p$wYb4ZFAmJ%}1)O
z8%18jX!jT0XD4&*n?yV?o;ERmXI{ZGJjE?&NTCA_SgB$U6S#sD!U)2HCw7rp3?cv%
zp4&tU7)2Vb2-qYibrf%{G~h9=VhCAiIE9^9wumg^K1#^qB#t7CW;pS}Eb<6rh#>^Q
zN0Z1jZej$dvCFhi#6hvXffq218_1#sd$Ao&SW_zJS=5Ic%8gxfcDq^`#cP+hY*i1b
z1MyTcalSj*-w}-Z|6A>eN=c6@bpgNHrw?j9+DNj$Gm(tb@^}5a<=@I^S&u54FJ{tH
znY2<c=VA$wGyLE<hhf}br1CjGT_W?ijUqD8aU34h!irfZ`_R#f{iwxiwa9xsL?3#w
z4GRvDI~c$L2R9X>@Zu{mS-=eP=tU2j(Fh-mNc|uxZ&1J`#1MuL--yW?R<VQ!7(zD=
z!+~XDGl@}X_JbnFD0V>bj@V4$I*d$dh`@(0oX`in#5~5WWyRcSDd}yBxioW5F&8fk
KySTv<Rlfk$7L?`y

delta 397
zcmXBQze@sf7{~GF>-|VOObs#!f?Gof6&l)X2uhG5!c{3~$Oye5Y{`RAngTiD{(~B9
z$P65csHv$z8iO*UA!u@l#@^5aFL;Lc^l@F&9T@SgWPUYgFU}{0i;ARUt?kEAC;lB#
zQ(Z0kq*TT0$ELf%$=`5uU|-pKCY&I(0{?x*ixu0oj3qJy10UleL)@W(I;vQJfniAG
z0cWVAfE-eI4~h)XL>4wa$3!mB#tHV2MH(ZM-jtfW@tmUt4-J&D422&<<QdoSu#X}N
z*gyhby2vX!I6@w?5WHw2xA5U$1#_5yXf^?R+`z*w*06*OtU$A$^2?h__wRR<?j2Sm
Ry~_i&9y)ePJ*T8j^*`QuMhO4_

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index 11e467ea2..d1fd38abd 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -926,6 +926,10 @@ def train_one_epoch(
 
     cur_batch_idx = params.get("cur_batch_idx", 0)
 
+    if params.multi_optim:
+        optimizer_enc, optimizer_dec = optimizer[0], optimizer[1]
+        scheduler_enc, scheduler_dec = scheduler[0], scheduler[1]
+
     for batch_idx, batch in enumerate(train_dl):
         if batch_idx < cur_batch_idx:
             continue