eval mode

3 years ago · 4e31a4445d
parent dc479df9e4
commit 4e31a4445d
1 changed files with 1 additions and 0 deletions
--- a/paddlespeech/s2t/exps/u2_st/model.py
+++ b/paddlespeech/s2t/exps/u2_st/model.py
@ -432,6 +432,7 @@ class U2STTester(U2STTrainer):
        """"E2E translation from extracted audio feature"""
        cfg = self.config.decoding
        text_feature = self.test_loader.collate_fn.text_feature
+        self.model.eval()

        hyps = self.model.decode(
            audio,