from local

2025-12-11 06:55:27 +00:00 · 2022-12-26 15:15:50 +09:00 · 2022-12-26 15:15:50 +09:00 · f961e91489
commit f961e91489
parent 7114f16a85
2 changed files with 1 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@ -1022,6 +1022,7 @@ def train_one_epoch(
            # in the batch and there is no normalization to it so far.
            loggin.info('1')
            scaler.scale(loss).backward()
+            loggin.info('2')

            if params.multi_optim and (batch_idx+1) % params.accum_grads == 0:
                set_batch_count(model, params.batch_idx_train)