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

Fix set_attr modify underly type (#47500)

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