From 53803202ccd457594d1ee86c581e0b14321c1e49 Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Tue, 29 Sep 2020 18:24:45 +0800 Subject: [PATCH] add rename var in block (#27705) --- python/paddle/fluid/dygraph/io.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/dygraph/io.py b/python/paddle/fluid/dygraph/io.py index 4391843b0e..4a3dacbd1a 100644 --- a/python/paddle/fluid/dygraph/io.py +++ b/python/paddle/fluid/dygraph/io.py @@ -139,6 +139,7 @@ def _append_loaded_suffix_to_var(program_desc): var_desc.set_name(new_name) for block_idx in six.moves.range(program_desc.num_blocks()): block = program_desc.block(block_idx) + block._rename_var(cpt.to_bytes(old_name), cpt.to_bytes(new_name)) for op_idx in six.moves.range(block.op_size()): op = block.op(op_idx) op._rename_input(old_name, new_name) -- GitLab