fix test

deepspeech/__init__.py

@@ -357,9 +357,9 @@ if not hasattr(paddle.Tensor, 'tolist'):
 ########### hcak paddle.nn.functional #############
 
 
-def glu(x: paddle.Tensor, dim=-1) -> paddle.Tensor:
+def glu(x: paddle.Tensor, axis=-1) -> paddle.Tensor:
     """The gated linear unit (GLU) activation."""
-    a, b = x.split(2, axis=dim)
+    a, b = x.split(2, axis=axis)
     act_b = F.sigmoid(b)
     return a * act_b
 

deepspeech/exps/u2/bin/test.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 """Evaluation for U2 model."""
 import cProfile
-import os
 
 from deepspeech.exps.u2.config import get_cfg_defaults
 from deepspeech.exps.u2.model import U2Tester as Tester
@@ -53,4 +52,4 @@ if __name__ == "__main__":
     # Setting for profiling
     pr = cProfile.Profile()
     pr.runcall(main, config, args)
-    pr.dump_stats(os.path.join(args.output, 'train.profile'))
+    pr.dump_stats('test.profile')

examples/aishell/s1/conf/conformer.yaml

@@ -91,7 +91,7 @@ training:
 decoding:
   batch_size: 128
   error_rate_type: cer 
-  decoding_method: attention  # 'attention', 'ctc_greedy_search', 'ctc_prefix_beam_search', 'attention_rescoring'
+  decoding_method: attention # 'attention', 'ctc_greedy_search', 'ctc_prefix_beam_search', 'attention_rescoring'
   lang_model_path: data/lm/common_crawl_00.prune01111.trie.klm
   alpha: 2.5
   beta: 0.3