From 6a091da0b0543befb0492848d3583700c274d111 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Wed, 23 Mar 2022 12:22:05 +0800 Subject: [PATCH] Minor fixes for saving checkpoints. (#265) * Minor fixes for saving checkpoints. * Fix loading checkpoints saved by previous code. --- egs/librispeech/ASR/pruned_transducer_stateless/train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless/train.py b/egs/librispeech/ASR/pruned_transducer_stateless/train.py index e71f0d1c6..0f51b4382 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless/train.py @@ -395,9 +395,10 @@ def load_checkpoint_if_available( "cur_batch_idx", ] for k in keys: - params[k] = saved_params[k] + params[k] = saved_params.get(k, 0) - params["start_epoch"] = saved_params["cur_epoch"] + if "cur_epoch" in saved_params: + params["start_epoch"] = saved_params["cur_epoch"] return saved_params