From 4c735f24ea954a5ced87bd0bd1573fa3eb914c18 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Thu, 13 Jun 2019 22:10:26 +0800 Subject: [PATCH] fix bug in fleet, test=develop (#18058) --- .../distribute_transpiler/__init__.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py b/python/paddle/fluid/incubate/fleet/parameter_server/distribute_transpiler/__init__.py index fcd42b66154..5b80bdb95d8 100644 --- 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) -- GitLab