diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp index b0e9e2056..f0855759e 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py index e0b272735..a688b12d3 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py @@ -717,6 +717,8 @@ def load_checkpoint_if_available( if "cur_batch_idx" in saved_params: params["cur_batch_idx"] = saved_params["cur_batch_idx"] + params.batch_idx_train = 0 + return saved_params