From c55ea2082f57c831ac12dc66bd7862afb263c2aa Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 27 Dec 2022 10:34:43 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 106496 -> 106496 bytes .../train.py | 3 +++ 2 files changed, 3 insertions(+) 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 66dc3d49859bbc1450f5d571c9b8227eed550f33..e88513ee1870cf3be9d5755d9af38ae16bc864c9 100644 GIT binary patch delta 192 zcmZoTz}9epO*F|M%+puFQqO<^2m}}ymU^#BPTD9sTZqx1d85$wjY5pUvb^s(7#L1- zLKGNG7c^i@PQ1m#z_1vIGk`c9h*g024>wR55YGkTIY3+m#M(ft1;m^{%mKvjxfmF( z0P$`ho(05}KpX+Yu0X5?#Oy%)258mk?N<#L|8q?LV8m!Yo!^L&b^CrR#t^2-8(%)z K9$?G3AqfCUWhy=Z delta 212 zcmZoTz}9epO*F|M%+puFQqO<^2m}}y=6kP7w%#Z@TZmD&d85$wjY5pUvb>8p85oMV zAPRJ+3mPycC-U(!Fr4OLU|0;qML_HV#Lhsh2*j7T85k}C@dhAX55&`exEhF)fY=d; z6@d6Z(6~21d=`k;0r6xY23cVS#Q%WSo!EZWfbl=a^bf|2iqjXEF>+20@E4ta!HiLP e`XpmUU67FU^aJLM8r%O{GUhUE7q(@5n*;!E^DuG% 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 01441b80b..c2665016f 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1562,6 +1562,9 @@ def run_adapter(rank, world_size, args, wb=None): valid_dl = librispeech.valid_dataloaders(valid_cuts) scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0) + if checkpoints and "grad_scaler" in checkpoints: + logging.info("Loading grad scaler state dict") + scaler.load_state_dict(checkpoints["grad_scaler"]) for epoch in range(params.start_epoch, params.num_epochs + 1): scheduler.step_epoch(epoch - 1)