From 3b140f3576925d7cb09b7b1112baf16500ddce96 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Wed, 29 Mar 2023 22:07:05 +0900
Subject: [PATCH] from local

---
 .../.train_adapter.py.swp                     | Bin 86016 -> 90112 bytes
 .../train_adapter.py                          |   2 ++
 2 files changed, 2 insertions(+)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp
index ca8de810f25713a12d113f59af91701285165259..63f255c61498cf6073138d323558b6865aab7463 100644
GIT binary patch
delta 476
zcmXZXJxD@P6bJBg)u&lL^@SozgJ$#rp+bQb4K*k<NDyQZWIa=w4AE=~lpt!AM}`d{
zeNeQ8!wXuBzG!F<B`64*BBO$!p#EY9e%y1<xfjmK6u3-@GlxCFk$}D5<spQTsN!vl
z1dh!w24*TMjXYz1Nf0c8xz<KF>j(W7t-`f_#h$Y>72mBUn$@6#|9)ApGYd;6aIe+o
zU>F9Wigza<!x7Ab7fkTZ5#50dS?Ggy_)!sk!ybe|1Rs1Vi9TQ-LeL96@T5RZK?aH-
zal#iPx@DV>j9H~hUN2QgHFDL#mO1I{*(McyEzNN;A;yxfg;Z=N6_1FE(WHFX!6rM?
zVuWZ-OXO~>@8<kpzHA`6gk9K%F|dIZba105%EJ<j!vI*}ohRCYbqK*UsNq6KbPfkF
u0YT8{sFg}hL^;?1KWN|z!>qy-xaGGY7S(1Sx+H$5K`P(t<yQwg<^BLAfLNgb

delta 238
zcmZoTz}m2YRW!*U%+puFQqO<^2m}}y&X}vD$ZiyU%g-phnOT6BkuiO;prH6>55XOL
z6CWrvDNIvfWZ_i=s=dhw(U87fP=QgLvA%_!fgvA=eSz2mi2txLFdPBm!$7<Zh>L;P
z8Hg2t_&+NH!xJDr48(JR*ae7<fS3b_*@1Wi3j;$Q5a$AMAP_SGF#`~9WM*K<0%8Xs
rwg=)5Oh7RpE?{EdW6%dOSb!Mhup5&XPE^^hAH{f)b^9uH#xM;4u;nR-

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
index deb05a488..59cc60828 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
@@ -1018,6 +1018,8 @@ def train_one_epoch(
         scheduler_enc, scheduler_dec = scheduler[0], scheduler[1]
 
     for batch_idx, batch in enumerate(train_dl):
+        if params.batch_idx_train > params.num_updates:
+            break
         if batch_idx < cur_batch_idx:
             continue
         cur_batch_idx = batch_idx