未验证 提交 86eafde9 编写于 作者: C Chen Weihang 提交者: GitHub

fix test jit save load failed (#40180)

上级 bcaf88d2
......@@ -885,7 +885,7 @@ def _run_dygraph(instance, input, program_holder):
'start_op_index': 0,
'end_op_index': end_op_index,
'is_test': instance._is_test,
'program_id': _hash_with_id(trace_program)
'program_id': _hash_with_id(trace_program, instance)
})
# NOTE: [ why need set param's gradient type here ]
# if user set sparse gradient mode, the param's gradient
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册