From c7bfa2c95ab6521952441ce369b7aa03b668b5e0 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 10 Dec 2022 13:59:03 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 98304 -> 98304 bytes .../train.py | 4 ++-- 2 files changed, 2 insertions(+), 2 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 1ffe56c1dd719891d448df3a5a72f4bee804398c..2afb5b7b4e1ddc96b2d00baf36b823b9825701b5 100644 GIT binary patch delta 414 zcmWN{K`2839LMqhtM}f#&CFs_F0Ynyv7T~4yX{~HN-c4aX0p=cMXQO~Biuwi^-i2u z4vM6d+AgBRT{tMch1$WD^8MB4*YEIi9NlsB*qk+)9y5m$mXLOlhv-r6ek*nvQ<}?K zMir@ghONXag@}lWMw^I_XL!gXg*aN^hehf*!!CBvkFQpdPds81WhCH-L~gK+G)B=2 z1Ab8C2?s%`NSU#L6b4`+jxgR0krTMEF#!u9{I!VuqKP}~VIAY>!B0Ts8pkMM8iR=7 zLl?QlsouwdQNk1~1n{kiyx|p9>}wZ_I-+`ZMp3=m0`*cuQBzIg*%jsRAAS5zBLDyZ delta 376 zcmWN{!7GDt0LJm>SFe|M*sRx34%9Ao(F$pCm(W^C(Ud5~#jC7P>wSekK$G|4q!q7& zP};#^%8B#J!No~ANXdzd?^B=p9-bhugTPKLXXhqo`g4O>u@31hdAxG6m%2-tuZ^a_ zk{Y4onY8aEq-k`*g^kC!RDp*)hT-C~U3$e0t}%~pbm6y6xZZxCeVlXX6Xr49AXJ$NaEX; zUhFgnqKX3)F@zq(;Gj_}Y?^WF=xr+WW{nj-tQxDbvz@5bOJ>cfMC+Nb?3?|6k&Zuu 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 1b460cab7..c6f410bd2 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -934,8 +934,8 @@ def train_one_epoch( if batch_idx < cur_batch_idx: continue cur_batch_idx = batch_idx - - params.batch_idx_train += 1 + + if batch_idx % params.accum_grads == 0: params.batch_idx_train += 1 batch_size = len(batch["supervisions"]["text"]) try: