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 84bae58a7a3eb500b67010f81cdcdb66a7d38dc6..8d6be81fb113a3eaa0b47e595691492c9012cf37 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):
         """