From 2ccf2705102e620e334abb590c7888307ed37550 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Fri, 21 May 2021 09:56:24 +0000 Subject: [PATCH] fix ds2 export audio shape from B,D,T to B,T,D --- deepspeech/exps/deepspeech2/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepspeech/exps/deepspeech2/model.py b/deepspeech/exps/deepspeech2/model.py index 04137419c..8e8a18245 100644 --- a/deepspeech/exps/deepspeech2/model.py +++ b/deepspeech/exps/deepspeech2/model.py @@ -272,8 +272,8 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): infer_model, input_spec=[ paddle.static.InputSpec( - shape=[None, feat_dim, None], - dtype='float32'), # audio, [B,D,T] + shape=[None, None, feat_dim], + dtype='float32'), # audio, [B,T,D] paddle.static.InputSpec(shape=[None], dtype='int64'), # audio_length, [B] ])