We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 11997c8 commit 2c5fee5Copy full SHA for 2c5fee5
python/paddle/fluid/io.py
@@ -948,10 +948,11 @@ def save_inference_model(dirname,
948
# TODO(Superjomn) add an IR pass to remove 1-scale op.
949
with program_guard(main_program):
950
uniq_target_vars = []
951
- for var in target_vars:
+ for i, var in enumerate(target_vars):
952
if isinstance(var, Variable):
953
- var1 = layers.scale(var, 1.)
954
- uniq_target_vars.append(var1)
+ var = layers.scale(
+ var, 1., name="save_infer_model/scale_{}".format(i))
955
+ uniq_target_vars.append(var)
956
target_vars = uniq_target_vars
957
958
# when a pserver and a trainer running on the same machine, mkdir may conflict
0 commit comments