Merge pull request #1065 from LittleChenCc/develop

[ST] activate eval mode in translation demo function
pull/1067/head
Hui Zhang 3 years ago committed by GitHub
commit ec9e70d542
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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):
""""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,

Loading…
Cancel
Save