提交 ec367659 编写于 作者: C Channingss

fix bug of hard_swish

上级 06c95839
...@@ -853,8 +853,9 @@ class OpSet9(object): ...@@ -853,8 +853,9 @@ class OpSet9(object):
node0 = helper.make_node( node0 = helper.make_node(
'Add', inputs=[op.input('X')[0], offset_name], outputs=[name0]) 'Add', inputs=[op.input('X')[0], offset_name], outputs=[name0])
name1 = self.get_name(op.type, 'relu') name1 = self.get_name(op.type, 'relu')
min_value = op.attr('min') min_value = 0.0
max_value = op.attr('max') max_value = op.attr('threshold')
node1 = helper.make_node( node1 = helper.make_node(
'Clip', 'Clip',
inputs=[name0], inputs=[name0],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册