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

---
 .../.train.py.swp                             | Bin 98304 -> 98304 bytes
 .../train.py                                  |   2 +-
 2 files changed, 1 insertion(+), 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 992764b00da73c8b7ad11332f6c5d82505da77f3..95cf2146fee677c67f3a76344f858c324eada5fc 100644
GIT binary patch
delta 134
zcmZo@U~6b#6HPJ*^Ym4))H7fJ0s#hwZow(Z*&9XQ+p|t&VqlQn%;}KKD9Opn!0?oX
zfngUA_XBY`5c>eJ?B-U-8BCLTU8E+*x_nSm$S+ce_6Z73&PYuu%}Ff^sVqp<Q9u)j
Sot!vTt-08Bd$B8HuN45m+$I(P

delta 103
zcmZo@U~6b#6HPJ*^Ym4))H7fJ0s#hwV!<iN&KpJF+p}(GVqkFI%;}KKD5=lNz`zZ}
z4_Fu&wgT~7AT9!8=gqB-Gngjpx^PXFb&+CHP}nT%!o@iGTaa3FvFr9?SH@l|0Jdxx
A(EtDd

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 3584067f4..d6d729239 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -878,7 +878,7 @@ def train_one_epoch(
     params: AttributeDict,
     model: Union[nn.Module, DDP],
     optimizer: torch.optim.Optimizer or [torch.optim.Optimizer, torch.optim.Optimizer],
-    scheduler: LRSchedulerType,
+    scheduler: LRSchedulerType or [LRSchedulerType, LRSchedulerType],
     sp: spm.SentencePieceProcessor,
     train_dl: torch.utils.data.DataLoader,
     valid_dl: torch.utils.data.DataLoader,