Use eval() for the masked LM model in decoding.

2025-08-26 18:24:18 +00:00 · 2021-11-16 10:55:14 +08:00 · 2021-11-16 10:55:14 +08:00 · b8037cd529
commit b8037cd529
parent 7f5d9a1671
1 changed files with 1 additions and 0 deletions
--- 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