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')