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

---
 .../.train.py.swp                             | Bin 90112 -> 90112 bytes
 .../train.py                                  |  12 +++++++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

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 1b5f62db46cfdf19e06ccf6beb8e9dc0e9213c4e..302494e07f07036cb640036a8be60cf307f2d37c 100644
GIT binary patch
delta 539
zcmZ9|Pbh<790%~<!#kVV#Au{e&qlElTXOI(ii?)!z;fB<^=|TFf10T!+KL=-kXPP_
z=H$W!haFsY#zCQQa#WnO6Bpl?l05Z!p5OEPJ%4(hOkS7C>#P$l-{27MXmJr*1yQPE
zTQa_%TI(##v+TCy+!l7*W3Ca1g1Vm-9<_3qb)}8vM6YI|GnfVk*r5uZOhkLI3!AVA
zqtF00cqt<~fn!($9}IvM?u|ruZ~+Qzfe7Q!3>J7Sr8FIu5@jI?jbMWZ1JM!e!w8t5
zpeMS5OW1&Q=!HTFa<Bs%Fby(v0fSGx_XDos079VsK>++Z9BfDMCY&>2uUK92v39NN
zF5*0gv0tx@WzFi1pS3A3OD4seDp4x~tX`2;tF(#`4231F+^E+ghu^7qvsNJ(j{al~
zT7)bq&r8+6yX!cbJjvr?LX6C{MCNCcaxfN8%8^dBHNa~7njGIVc_Ex|a5>7QsjxH?
YkIB*GoRh~A{7?P+g8yrBVt{4Y7xlz>YXATM

delta 349
zcmXBPPbh<N9LMqRPkZ(U&zdJaH4awFQO^#N#$~m`ilUa57Q(?q99+!7vQaJ^wt6^N
zxwx`Mk5Y^{xsbC`j@0bvZ{FoouiNMH&Q`T-RWqlo(fClPuiFw95Sb4Yr(f!2v&~Sk
zT-b2JhI$;(6Pifg{r`yBEyF6ZX8A?FbdgJ3;1mbQAdV<>JT!~k;|QxrBZAK+kq<oM
z3=TH2ff4lLw^8I4H`v1<qWJO2gxvW=uCRhIf_V3e)KJDTW)MWpBXW*1cCd{}bRmQe
z)akp%7V_?ja*b*BTSE0zyz5U5_DNE8I>n^&yX?c1>MzWdO#3^f+6()oR?p(na(W@1
NDZJ)_&g{6_Re#jeNiqNc

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 b0e8a02b5..305ebd8bb 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1178,9 +1178,15 @@ def run(rank, world_size, args, wb=None):
 
         scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs)
 
-    if checkpoints and "optimizer" in checkpoints:
-        logging.info("Loading optimizer state dict")
-        optimizer.load_state_dict(checkpoints["optimizer"])
+    if checkpoints and ("optimizer" in checkpoints) or ("optimizer_enc" in checkpoints):
+        if params.multi_optim:
+            logging.info("Loading optimizer state dict")
+            optimizer_enc.load_state_dict(checkpoints["optimizer_enc"])
+            optimizer_dec.load_state_dict(checkpoints["optimizer_dec"])
+
+        else:
+            logging.info("Loading optimizer state dict")
+            optimizer.load_state_dict(checkpoints["optimizer"])
 
     if (
         checkpoints