align the infer_model format of Paddle2.0 (#85)

Co-authored-by: majun16 <majun16@baidu.com>
qingshui · Jul 13, 2023 · 637c5b9 · 637c5b9
1 parent 4c89fdb
commit 637c5b9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py
@@ -1369,6 +1369,9 @@ def save_inference_model(dirname,
     with program_guard(main_program):
         uniq_target_vars = []
         for i, var in enumerate(target_vars):
+            if isinstance(var, Variable):
+                var = layers.scale(
+                    var, 1., name="save_infer_model/scale_{}".format(i))
             uniq_target_vars.append(var)
         target_vars = uniq_target_vars
     target_var_name_list = [var.name for var in target_vars]