未验证 提交 fe9834cf 编写于 作者: Webbley's avatar Webbley 提交者: GitHub

Merge pull request #48 from Liwb5/develop

fix bug in gin layer when activation function is None
......@@ -239,7 +239,8 @@ def gin(gw,
bias_attr=fluid.ParamAttr(name="%s_b_0" % name))
output = fluid.layers.batch_norm(output)
output = getattr(fluid.layers, activation)(output)
if activation is not None:
output = getattr(fluid.layers, activation)(output)
output = fluid.layers.fc(output,
size=hidden_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册