提交 4e31a444 编写于 作者: J Junkun

eval mode

上级 dc479df9
...@@ -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.
先完成此消息的编辑!
想要评论请 注册