未验证 提交 ec9e70d5 编写于 作者: H Hui Zhang 提交者: GitHub

Merge pull request #1065 from LittleChenCc/develop

[ST] activate eval mode in translation demo function
...@@ -86,7 +86,7 @@ training: ...@@ -86,7 +86,7 @@ training:
scheduler_conf: scheduler_conf:
warmup_steps: 25000 warmup_steps: 25000
lr_decay: 1.0 lr_decay: 1.0
log_interval: 5 log_interval: 50
checkpoint: checkpoint:
kbest_n: 50 kbest_n: 50
latest_n: 5 latest_n: 5
......
...@@ -432,6 +432,7 @@ class U2STTester(U2STTrainer): ...@@ -432,6 +432,7 @@ class U2STTester(U2STTrainer):
""""E2E translation from extracted audio feature""" """"E2E translation from extracted audio feature"""
cfg = self.config.decoding cfg = self.config.decoding
text_feature = self.test_loader.collate_fn.text_feature text_feature = self.test_loader.collate_fn.text_feature
self.model.eval()
hyps = self.model.decode( hyps = self.model.decode(
audio, audio,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册