未验证 提交 6045c7b5 编写于 作者: O OuYang Yu 提交者: GitHub

Merge pull request #137 from Oneflow-Inc/dev_fix_bn_relu_false

fix fuse_bn_relu is false
......@@ -102,8 +102,7 @@ class ResnetBuilder(object):
name=name + "_bn_relu",
)
else:
output = self._batch_norm(output, name + "_bn", last=last)
return flow.nn.relu(output)
return flow.nn.relu(self._batch_norm(inputs, name + "_bn", last=last))
def _batch_norm_add_relu(self, inputs, addend, name=None, last=False):
if self.fuse_bn_add_relu:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册