From 86eafde9d3c9b9e95f9f7b3594a86217aa2c1a55 Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Sat, 5 Mar 2022 08:34:37 +0800 Subject: [PATCH] fix test jit save load failed (#40180) --- python/paddle/fluid/dygraph/io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/dygraph/io.py b/python/paddle/fluid/dygraph/io.py index aad77373509..f58952d3036 100644 --- a/python/paddle/fluid/dygraph/io.py +++ b/python/paddle/fluid/dygraph/io.py @@ -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 -- GitLab