diff --git a/paddle_fl/examples/submitter_demo/scheduler_client.py b/paddle_fl/examples/submitter_demo/scheduler_client.py index b19c1793e401e473536c72f22fbbc1fe30d7d3a5..eff68df845129feaca093b4277f1970dcfb911e9 100644 --- a/paddle_fl/examples/submitter_demo/scheduler_client.py +++ b/paddle_fl/examples/submitter_demo/scheduler_client.py @@ -18,8 +18,8 @@ print(random_port) current_ip = socket.gethostbyname(socket.gethostname()) endpoints = "{}:{}".format(current_ip, random_port) #start a web server for remote endpoints to download their config -#os.system("python -m SimpleHTTPServer 8080 &") -os.system("python -m http.server 8080 &") +os.system("python -m SimpleHTTPServer 8080 &") +#os.system("python -m http.server 8080 &") if os.path.exists("job_config"): os.system("rm -rf job_config") if os.path.exists("package"): diff --git a/paddle_fl/examples/submitter_demo/train_program.py b/paddle_fl/examples/submitter_demo/train_program.py index 22cef9f92012dc816906431a5671994f5fbb3363..a5f10fab0eb2fe8aae506f71f4961c5a5ea5cdfa 100644 --- a/paddle_fl/examples/submitter_demo/train_program.py +++ b/paddle_fl/examples/submitter_demo/train_program.py @@ -91,12 +91,12 @@ else: output_folder = "fl_model" epoch_id = 0 while not trainer.stop(): - print("batch %d start train" % (step_i)) + print("epoch %d start train" % (epoch_id)) step_i = 0 for data in reader(): trainer.run(feed=data, fetch=[]) step_i += 1 - if train_step == trainer._step: + if step_i == trainer._step: break epoch_id += 1 if epoch_id % 5 == 0: diff --git a/paddle_fl/version.py b/paddle_fl/version.py index 1ba4b84c2ffe81085dea2f8813a922be92b980d2..2f9278ce368bcbfc0d65b147d84fcd2938414638 100644 --- a/paddle_fl/version.py +++ b/paddle_fl/version.py @@ -12,6 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. """ PaddleFL version string """ -fl_version = "0.1.7" -module_proto_version = "0.1.7" +fl_version = "0.1.10" +module_proto_version = "0.1.10" diff --git a/setup.py b/setup.py index c5e3baf07e79091ce708075b40e9a748564385f1..ba29a1496d39949b84f3b74eeec935d831e57c01 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', 'zmq' + 'six >= 1.10.0', 'protobuf >= 3.1.0','paddlepaddle >= 1.6', 'zmq', 'paddlepaddle-gpu >= 1.6' ] if max_version < 3: