fix bug in fleet, test=develop (#18058)

4c735f24 · tangwei12 · GitHub · 42f12a4a · 4c735f24
隐藏空白更改
内联并排

Showing with 3 addition and 10 deletion

python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py .../fleet/parameter_server/distribute_transpiler/__init__.py +3 -10

未找到文件。
--- a/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
+++ b/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py
@@ -144,16 +144,9 @@ class DistributedTranspiler(Fleet):
                                    executor, main_program, None, None,
                                    export_for_deployment)
        else:
-            io.save_inference_model(
+            io.save_inference_model(dirname, feeded_var_names, target_vars,
-                dirname,
+                                    executor, self._origin_program, None, None,
-                feeded_var_names,
+                                    export_for_deployment, True)
-                target_vars,
-                executor,
-                self._origin_program,
-                None,
-                None,
-                export_for_deployment,
-                model_only=True)
            model_basename = "__model__"
            model_filename = os.path.join(dirname, model_basename)