From 136c03d040ad8d81ae8d0ccf5a3a5d9d11d9c79c Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Wed, 9 Feb 2022 12:15:12 +0800 Subject: [PATCH] Fix decoding. --- egs/librispeech/ASR/transducer_stateless/decode.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/transducer_stateless/decode.py b/egs/librispeech/ASR/transducer_stateless/decode.py index c101d9397..0c07ba308 100755 --- a/egs/librispeech/ASR/transducer_stateless/decode.py +++ b/egs/librispeech/ASR/transducer_stateless/decode.py @@ -441,7 +441,9 @@ def main(): filenames.append(f"{params.exp_dir}/epoch-{i}.pt") logging.info(f"averaging {filenames}") model.to(device) - model.load_state_dict(average_checkpoints(filenames, device=device)) + model.load_state_dict( + average_checkpoints(filenames, device=device), strict=False + ) model.to(device) model.eval()