[Hackathon 7th] 修复 tal_cs 测试中 0D tensor to 1D (#3913)

* [Fix] 0D tensor to 1D * [Update] feat dim
PaddlePaddle · Nov 29, 2024 · 5e8c727 · 5e8c727
1 parent a397ebe
commit 5e8c727
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/paddlespeech/s2t/exps/deepspeech2/bin/test_wav.py b/paddlespeech/s2t/exps/deepspeech2/bin/test_wav.py
@@ -75,7 +75,7 @@ def test(self):
         feat = self.preprocessing(audio, **self.preprocess_args)
         logger.info(f"feat shape: {feat.shape}")
 
-        audio_len = paddle.to_tensor(feat.shape[0])
+        audio_len = paddle.to_tensor(feat.shape[0]).unsqueeze(0)
         audio = paddle.to_tensor(feat, dtype='float32').unsqueeze(axis=0)
 
         result_transcripts = self.compute_result_transcripts(

diff --git a/paddlespeech/s2t/exps/u2/bin/quant.py b/paddlespeech/s2t/exps/u2/bin/quant.py
@@ -75,7 +75,7 @@ def run(self):
                     feat = self.preprocessing(audio, **self.preprocess_args)
                     logger.info(f"feat shape: {feat.shape}")
 
-                    ilen = paddle.to_tensor(feat.shape[0])
+                    ilen = paddle.to_tensor(feat.shape[0]).unsqueeze(0)
                     xs = paddle.to_tensor(feat, dtype='float32').unsqueeze(0)
                     decode_config = self.config.decode
                     logger.info(f"decode cfg: {decode_config}")

diff --git a/paddlespeech/s2t/exps/u2/bin/test_wav.py b/paddlespeech/s2t/exps/u2/bin/test_wav.py
@@ -78,7 +78,7 @@ def run(self):
             if self.args.debug:
                 np.savetxt("feat.transform.txt", feat)
 
-            ilen = paddle.to_tensor(feat.shape[0])
+            ilen = paddle.to_tensor(feat.shape[0]).unsqueeze(0)
             xs = paddle.to_tensor(feat, dtype='float32').unsqueeze(0)
             decode_config = self.config.decode
             logger.info(f"decode cfg: {decode_config}")