Skip to content

Commit 2c5fee5

Browse files
fc500110Yibing Liu
authored andcommitted
save infer model bug fix (#16224)
* save infer model bug fix * test=release/1.3
1 parent 11997c8 commit 2c5fee5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

python/paddle/fluid/io.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -948,10 +948,11 @@ def save_inference_model(dirname,
948948
# TODO(Superjomn) add an IR pass to remove 1-scale op.
949949
with program_guard(main_program):
950950
uniq_target_vars = []
951-
for var in target_vars:
951+
for i, var in enumerate(target_vars):
952952
if isinstance(var, Variable):
953-
var1 = layers.scale(var, 1.)
954-
uniq_target_vars.append(var1)
953+
var = layers.scale(
954+
var, 1., name="save_infer_model/scale_{}".format(i))
955+
uniq_target_vars.append(var)
955956
target_vars = uniq_target_vars
956957

957958
# when a pserver and a trainer running on the same machine, mkdir may conflict

0 commit comments

Comments
 (0)