From 3638320f3b75e3949ffe77a546a918a0bc4fb295 Mon Sep 17 00:00:00 2001 From: Jackwaterveg <87408988+Jackwaterveg@users.noreply.github.com> Date: Fri, 20 May 2022 20:25:46 +0800 Subject: [PATCH] fix self.max_len --- paddlespeech/cli/asr/infer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py index eb9721206..8b10b6b65 100644 --- a/paddlespeech/cli/asr/infer.py +++ b/paddlespeech/cli/asr/infer.py @@ -135,6 +135,8 @@ class ASRExecutor(BaseExecutor): Init model and other resources from a specific path. """ logger.info("start to init the model") + # default max_len: unit:second + self.max_len = 50 if hasattr(self, 'model'): logger.info('Model had been initialized.') return @@ -203,8 +205,6 @@ class ASRExecutor(BaseExecutor): self.model.set_state_dict(model_dict) # compute the max len limit - # default max_len: unit:second - self.max_len = 50 if "conformer" in model_type or "transformer" in model_type or "wenetspeech" in model_type: # in transformer like model, we may use the subsample rate cnn network subsample_rate = self.model.subsampling_rate()