From f1178e9d797db5186293a70b14213be1cfcac4cc Mon Sep 17 00:00:00 2001
From: xujiaqi01 <173596896@qq.com>
Date: Thu, 28 Nov 2019 14:21:44 +0800
Subject: [PATCH] fix fleet save bug (#21362)

* fix fleet save bug of save_infernece_model
* test=develop
---
 .../fluid/incubate/fleet/parameter_server/pslib/__init__.py     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/fluid/incubate/fleet/parameter_server/pslib/__init__.py b/python/paddle/fluid/incubate/fleet/parameter_server/pslib/__init__.py
index 84bae58a7a..8d6be81fb1 100644
--- a/python/paddle/fluid/incubate/fleet/parameter_server/pslib/__init__.py
+++ b/python/paddle/fluid/incubate/fleet/parameter_server/pslib/__init__.py
@@ -236,7 +236,7 @@ class PSLib(Fleet):
               fleet.save_inference_model(dirname="hdfs:/my/path")
 
         """
-        self._fleet_ptr.save_model(dirname)
+        self._fleet_ptr.save_model(dirname, 0)
 
     def print_table_stat(self, table_id):
         """
-- 
GitLab