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

6 years ago · 4c735f24ea
parent 42f12a4aca
commit 4c735f24ea
1 changed files with 3 additions and 10 deletions
--- 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(
-                dirname,
-                feeded_var_names,
-                target_vars,
-                executor,
-                self._origin_program,
-                None,
-                None,
-                export_for_deployment,
-                model_only=True)
+            io.save_inference_model(dirname, feeded_var_names, target_vars,
+                                    executor, self._origin_program, None, None,
+                                    export_for_deployment, True)

            model_basename = "__model__"
            model_filename = os.path.join(dirname, model_basename)