fix fleet save bug (#21362)

* fix fleet save bug of save_infernece_model * test=develop
5 years ago · f1178e9d79
parent e2c6f434ec
commit f1178e9d79
1 changed files with 1 additions and 1 deletions
--- 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):
        """