diff --git a/paddle_fl/core/master/fl_job.py b/paddle_fl/core/master/fl_job.py index 955f8891ceb516bdc2e28afbef35a0c4bdae348d..a221897d38393cfb8998ea0d988cff4f3c22e61e 100644 --- a/paddle_fl/core/master/fl_job.py +++ b/paddle_fl/core/master/fl_job.py @@ -113,13 +113,13 @@ class FLCompileTimeJob(FLJobBase): self._save_readable_program( server_main, "%s/server.main.program.txt" % server_folder) - self._save_str_list(self._feed_names, + self._save_str_list(self._feed_names, "%s/feed_names" % server_folder) - self._save_str_list(self._target_names, + self._save_str_list(self._target_names, "%s/target_names" % server_folder) - self._save_endpoints(self._server_endpoints, + self._save_endpoints(self._server_endpoints, "%s/endpoints" % server_folder) - self._save_strategy(self._strategy, + self._save_strategy(self._strategy, "%s/strategy.pkl" % server_folder) for i in range(trainer_num): @@ -139,7 +139,7 @@ class FLCompileTimeJob(FLJobBase): self._save_readable_program( trainer_main, "%s/trainer.main.program.txt" % trainer_folder) - self._save_str_list(self._feed_names, + self._save_str_list(self._feed_names, "%s/feed_names" % trainer_folder) self._save_str_list(self._target_names, "%s/target_names" % trainer_folder) diff --git a/setup.py b/setup.py index 74046178118e35ef06467d3893677feac2a18f62..c5e3baf07e79091ce708075b40e9a748564385f1 100644 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ def python_version(): max_version, mid_version, min_version = python_version() REQUIRED_PACKAGES = [ - 'six >= 1.10.0', 'protobuf >= 3.1.0','paddlepaddle >= 1.6' + 'six >= 1.10.0', 'protobuf >= 3.1.0','paddlepaddle >= 1.6', 'zmq' ] if max_version < 3: