未验证 提交 e6a6b0fc 编写于 作者: A Aurelius84 提交者: GitHub

Fix set_attr modify underly type (#47500)

上级 974f8f32
...@@ -6100,7 +6100,7 @@ class Program(object): ...@@ -6100,7 +6100,7 @@ class Program(object):
for j in range(block.op_size()): for j in range(block.op_size()):
op = block.op(j) op = block.op(j)
if op.has_attr('is_test'): if op.has_attr('is_test'):
op._set_attr('is_test', True) op._set_bool_attr('is_test', True)
if op.type() == "batch_norm": if op.type() == "batch_norm":
# Remove the output ReserveSpace of batch_norm if exists. # Remove the output ReserveSpace of batch_norm if exists.
op.remove_output("ReserveSpace") op.remove_output("ReserveSpace")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册