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

---
 .../.train.py.swp                             | Bin 110592 -> 110592 bytes
 .../train.py                                  |   1 +
 2 files changed, 1 insertion(+)

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 b61e611bd8ee54ae125f4488b3b9b87873815e8c..8236023747cd63ca42debe18d1aa7e23527fc098 100644
GIT binary patch
delta 150
zcmZp8z}E19ZNqzJMuz4O%-cUOGm6Y%HREAmI5wSg0i(SW6CVQuBM>tH@d;iAhV?)^
z2Z&pNI1h-ufY=3ywSia&h#7$REl~SeAU+N<hG(;)ur1Hz56&#xU#?*cV3f-%&5bWD
jNJ%V7Emp8q$jmFzNJ=b8&WKOWFU>2_+-|v!@i-#@XIv(1

delta 119
zcmZp8z}E19ZNqzJ#?a;u%-cUOGm6Y%4dG#6cru-H0i(T>0v`i|JP^wP@iSfqhO<Dt
z3y2p3aSsq@0C5r!djhcm5X%8EHxR!D8u1il49{jo;b@-8ADmgXpI*Zlz_{IV9piCE
E0Qg-TmH+?%

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 f8149c7ed..096995340 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -106,6 +106,7 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
     for module in model.modules():
         if hasattr(module, "batch_count"):
             module.batch_count = batch_count
+    model.module.encoder.num_updates = int(batch_count)
 
 
 def add_rep_arguments(parser: argparse.ArgumentParser):