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

---
 .../.train.py.swp                             | Bin 90112 -> 90112 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 60b856dab77ecf42dbb2e9d3009e061be0b323a2..6b4aa53a138697cf7fdab2e53f5b842df17ccc8c 100644
GIT binary patch
delta 263
zcmZoTz}j$tRV>LM%+puFQqO<^2m}}yqy?rVn|$0T_Dhh_f3u*_=UiSt76yi8tPolM
z$$|~*>$h_;Ff0J#P9W|8;zA&f0^&d*Rt91QAl}Q#z%UbtOM%!8h?Ri&9|r@&Um$)4
z#QT7FF%Z`RaRd;n0<ky{ivckQ5T9aaV3-BOSwO4`#1GgQ814gcH5-F4gE)}!1n8uV
zK)eBnm$7aRY7AqUyt!3!Gh5peW;O*NNX?sE*(o>qRp*4sg<UpWd5O8H#R|3x(Xo>k
QbV*E3oF~5dT$j;R0Dm<)EdT%j

delta 305
zcmZoTz}j$tRV>LM%+puFQqO<^2m}}yO8KWGulcZ1?3W;;_hvz%&$+ydSr`~zvqEIO
zCkr;LuV2W;z_0*_JAt?eh+TkK5s2?`GBE4{;%*>L2V!R+b^>B;AZ7;Q7aR->hk$r4
z5N82#3=l^Hu`dww1Mz)!28Q)OoCU<HK&%492iSn>fLIfVzX4tGm6bsl?3@ii&H|ud
z8xZ@lZ4PP-V-ZZvQ9yvyoZ{5!iE)g=>L4)?&x(NoCS9DIk(yEpR-Ouz1}PTbe6;le
b^W>AA6DB8h8Bbo*#XUJ`p4jGdT}D>{4j@CW

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 735210c61..f07c5663b 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1130,13 +1130,17 @@ def run(rank, world_size, args, wb=None):
     
     if params.multi_optim:
         logging.info("Using seperate optimizers over encoder, decoder ...")
+
         enc_param = []
+        enc_names = []
 
         dec_names = []
         dec_param = []
+        
         for n, p in model.named_parameters():
             name = n.split('.')[1]
             if name == 'encoder' and 'feature_extractor' not in n:
+                enc_names.append(n)
                 enc_param.append(p)
             elif 'feature_extractor' not in n:
                 dec_names.append(n)