From eca66f317ecd8a4e255c205f4417736634b2b49b Mon Sep 17 00:00:00 2001 From: xujiaqi01 <173596896@qq.com> Date: Thu, 21 Nov 2019 14:05:06 +0800 Subject: [PATCH] fix fleet util bug (#21254) * fix fleet util bug in save paddle inference model * test=develop --- python/paddle/fluid/incubate/fleet/utils/fleet_util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/incubate/fleet/utils/fleet_util.py b/python/paddle/fluid/incubate/fleet/utils/fleet_util.py index 30752b9e442..305043c58b6 100644 --- a/python/paddle/fluid/incubate/fleet/utils/fleet_util.py +++ b/python/paddle/fluid/incubate/fleet/utils/fleet_util.py @@ -919,7 +919,7 @@ class FleetUtil(object): dirname=model_name, feeded_var_names=feeded_var_names, target_vars=target_vars, - executor=exe, + executor=executor, main_program=program, params_filename="params") else: @@ -927,7 +927,7 @@ class FleetUtil(object): dirname=model_name, feeded_var_names=feeded_var_names, target_vars=target_vars, - executor=exe, + executor=executor, main_program=program) configs = { -- GitLab