fix ds2 export audio shape from B,D,T to B,T,D

4 years ago · 2ccf270510
parent 5d4f3fbd7b
commit 2ccf270510
1 changed files with 2 additions and 2 deletions
--- 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]
            ])