From 122422c5eb90317d3f5acf5255482ae9637ccdcf Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 15:02:58 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 110592 -> 110592 bytes
 .../train.py                                  |  15 ++++++++++++---
 2 files changed, 12 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 19136fd34c1e874bb19b29dbeea3b5134bdc9b4b..e0f87d3a1be062b1699e2af7668aec4f01054ea6 100644
GIT binary patch
delta 344
zcmZp8z}E19O*F|M%+puFQqO<^2m}}yniQrapV%mBEyU=yIZ$YGEUz^a14BMDMAB`t
zV8QBoX$}U4kL(N#vw^r3h@*j+6^P%nF)+LX;u}D`4u}(g*b9g~fmi{EZ?G~j90TG!
zAoc=c1t9*y!oY9_i01%t8xU6lu@4ac1KPL&hzo!iWP1cq%nyj2SQs{27n(3_ep>RH
zY4Qg>mdOHTQj?#RyMfs{6-twntE6G#N=hX~iJ5u&DXGcvIjM<7d6{|X@kNOxsY*Hu
v$)!c{K=DZxI&htSnH(U~V7gNCu<HVfBkMX-#y9!eT)}3yitTO{jJwPMM-N=|

delta 226
zcmZp8z}E19O*F|M%+puFQqO<^2m}}yvJ|EyFW4w*EyQTMIZ$YGEbmoj28O3B5J}s~
zf(5JVWjGiZF0nH(9020|Ks*hI{eaj8h@Y?l#esMl5N83g6A%jl@gG(OhW9{x2#8C7
zSPh7u0Zj#&Tn}V>0I@0%e**IN191-!cLQ-35GMd}1Q7cHu@ei!X6r%|rtNxNjAe|Q
i^NU|IO?Jo>n=D(-KDoZkesX@fQnO#hcE1Y7P38cErZ8^+

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 e9febaad0..f8149c7ed 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1064,9 +1064,18 @@ def train_one_epoch(
                 )
 
             if tb_writer is not None:
-                tb_writer.add_scalar(
-                    "train/learning_rate", cur_lr, params.batch_idx_train
-                )
+                if params.multi_optim:
+                    tb_writer.add_scalar(
+                        "train/enc_learning_rate", cur_enc_lr, params.batch_idx_train
+                    )
+                    tb_writer.add_scalar(
+                        "train/dec_learning_rate", cur_dec_lr, params.batch_idx_train
+                    )
+
+                else:
+                    tb_writer.add_scalar(
+                        "train/learning_rate", cur_lr, params.batch_idx_train
+                    )
 
                 loss_info.write_summary(
                     tb_writer, "train/current_", params.batch_idx_train