提交 5972955f 编写于 作者: 小湉湉's avatar 小湉湉

fix glu

上级 b874d846
...@@ -362,7 +362,7 @@ class GLU(nn.Layer): ...@@ -362,7 +362,7 @@ class GLU(nn.Layer):
self.dim = dim self.dim = dim
def forward(self, xs): def forward(self, xs):
return F.glu(xs, dim=self.dim) return F.glu(xs, axis=self.dim)
if not hasattr(paddle.nn, 'GLU'): if not hasattr(paddle.nn, 'GLU'):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册