bugfix: audio_len should be 1D, no 0D, which will raise list index out

of range error in the following decode process
pull/3490/head
Luzhenhui 2 years ago
parent 17854a04d4
commit f7fc54bade

@ -274,7 +274,7 @@ class ASRExecutor(BaseExecutor):
# fbank
audio = preprocessing(audio, **preprocess_args)
audio_len = paddle.to_tensor(audio.shape[0]).unsqueeze(axis=0)
audio_len = paddle.to_tensor([audio.shape[0]]).unsqueeze(axis=0)
audio = paddle.to_tensor(audio, dtype='float32').unsqueeze(axis=0)
self._inputs["audio"] = audio

Loading…
Cancel
Save