From 685beeb9e96ef614bf4969d22faf7014f893c2a8 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 26 Dec 2022 15:18:00 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 102400 -> 102400 bytes .../train.py | 12 ++++++++---- 2 files changed, 8 insertions(+), 4 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 dc672b2a1ce70e62f4549ef07885dcb9492e86b9..f8e0962d11510702e0b9d502c45c095e6c0091d5 100644 GIT binary patch delta 285 zcmZozz}B#UO*F|M%+puFQqO<^2m}}yf^1eMTW%DcEy$?dyisubMnT5lJYHi?28Q22 z3{s#yU9f@Cy55%;$OGbcJPZuSfp{qpw*ql85ZeN=G!RPxF%u9U;$~pj0L1HoxEzR6 zfH)b5!+_WUh(&?;A{PV0F(6(H#4SLa48(RoYyiaiK&%ABU?+gw$-uCgi^YI*^8Jq- z+q=6M&6t=h)F=BNm7adDhf#L=zFtN?CUf=46ORZ_RyZOL;hRDD4oB1=d{Z!gT0bMl WH4g*BQ6OFn#EXEq3y8gd*aC>nf%rc+1H&62ehtL;fcOLu z&jsR0Aoc@dZ6M|W;@3dqj{)%xAl?qdtAMzfYcnIuMb63hKXOdYn##4Ew~JASX?k`q gqsVlHXhzQIVLgnpljV=>pPt{tD75`dFQW+)0Cu=A1^@s6 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 afd8713e6..16792323e 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1036,11 +1036,15 @@ def train_one_epoch( elif not params.multi_optim and (batch_idx+1) % params.accum_grads == 0: logging.info('3') set_batch_count(model, params.batch_idx_train) - scheduler.step_batch(params.batch_idx_train) - scaler.step(optimizer) - scaler.update() - optimizer.zero_grad() logging.info('4') + scheduler.step_batch(params.batch_idx_train) + logging.info('5') + scaler.step(optimizer) + logging.info('6') + scaler.update() + logging.info('7') + optimizer.zero_grad() + logging.info('8') except: # noqa display_and_save_batch(batch, params=params, sp=sp)