From b598ede5e4cf8bdc71363d4c67e56c252f30d01e Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 14:51:49 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 110592 -> 110592 bytes
 .../train.py                                  |   6 +++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

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 02c929e2a201ecf8f023d72faa00eb6ac1792505..f3b36ecfbebcbfa3cd56110bec03f8aaae6c63fa 100644
GIT binary patch
delta 249
zcmZp8z}E19O*F|M%+puFQqO<^2m}}yIOV1!_iPm1EySp^`J#|<E$;(P28K0U5J{cM
zf)l*!kMJ@uOabB)APxg!Eg=5M!@zJFh))6W3LuUI;s_u%2VyfImIY!aAb!csz;FVH
zmjiJd5Elb+2oReBu@n&h1ln{Bi1!0A$ks<dd=iM)0r6BIF5zO>e0#!0&ds*7+nA<X
zMKSVCK08-x`owvRVw3G=aZW!~#mF`J(p=HWX0rsRb5}z+)6-%Z<(r?+-Tr(oqofc3
Da&$Ya

delta 216
zcmZp8z}E19O*F|M%+puFQqO<^2m}}y4$4kRmfk43TZmC(^F<-!THZ!31_oPhh@{43
z!3o~=yLlNHx_~$ah@*kn5{O^%FfhCX;uAo89EjHe@nj&b0^&#@wg6&rAZ7vLr$Ez=
z0r6@e?gHX;APxXxkS(r2tN_Gcfwn#d;zvMy5{TCU@l+r#;bPc)d%{J|Npn=EH|8_u
YY+f^~jcKykT$SeMbGJXA%P1)X0FGlX#sB~S

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 b59d065fb..ad95d67bb 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1302,7 +1302,11 @@ def run(rank, world_size, args, wb=None):
         scaler.load_state_dict(checkpoints["grad_scaler"])
 
     for epoch in range(params.start_epoch, params.num_epochs + 1):
-        scheduler.step_epoch(epoch - 1)
+        if params.multi_optim:
+            scheduler_enc.step_epoch(epoch - 1)
+            scheduler_dec.step_epoch(epoch - 1)
+        else:
+            scheduler.step_epoch(epoch - 1)
         fix_random_seed(params.seed + epoch - 1)
         train_dl.sampler.set_epoch(epoch - 1)