Merge pull request #792 from yt605155624/fix_glu

Fix glu
3 years ago · 1f53d54fdd
parent 8e8ad86cdf 5972955f62
commit 1f53d54fdd
1 changed files with 1 additions and 1 deletions
--- a/deepspeech/init.py
+++ b/deepspeech/init.py
@ -362,7 +362,7 @@ class GLU(nn.Layer):
        self.dim = dim

    def forward(self, xs):
-        return F.glu(xs, dim=self.dim)
+        return F.glu(xs, axis=self.dim)


 if not hasattr(paddle.nn, 'GLU'):