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

Merge pull request #48 from Liwb5/develop

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