From 5972955f6207f1627ac2f63188b9c4c258f1d8e5 Mon Sep 17 00:00:00 2001
From: TianYuan <white-sky@qq.com>
Date: Fri, 27 Aug 2021 13:27:11 +0000
Subject: [PATCH] fix glu

---
 deepspeech/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepspeech/__init__.py b/deepspeech/__init__.py
index fbec5a5e..d85a3dde 100644
--- 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'):
-- 
GitLab