From b8037cd529410eddc9e3d01ec259cbf604b07787 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Tue, 16 Nov 2021 10:55:14 +0800 Subject: [PATCH] Use eval() for the masked LM model in decoding. --- egs/librispeech/ASR/conformer_ctc/decode.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/conformer_ctc/decode.py b/egs/librispeech/ASR/conformer_ctc/decode.py index 1ffef3bff..0bdf426c2 100755 --- a/egs/librispeech/ASR/conformer_ctc/decode.py +++ b/egs/librispeech/ASR/conformer_ctc/decode.py @@ -754,6 +754,7 @@ def main(): ) masked_lm_model.to(device2) masked_lm_model.device = device2 + masked_lm_model.eval() else: masked_lm_model = None