Merge pull request #1065 from LittleChenCc/develop

[ST] activate eval mode in translation demo function
3 years ago · ec9e70d542
parent dc479df9e4 1f3357f2d2
commit ec9e70d542
2 changed files with 2 additions and 1 deletions
--- a/examples/ted_en_zh/st0/conf/transformer_mtl_noam.yaml
+++ b/examples/ted_en_zh/st0/conf/transformer_mtl_noam.yaml
@ -86,7 +86,7 @@ training:
  scheduler_conf:
    warmup_steps: 25000
    lr_decay: 1.0
-  log_interval: 5
+  log_interval: 50
  checkpoint:
    kbest_n: 50
    latest_n: 5
--- 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,