diff --git a/paddle_fl/examples/generate_job_from_program/fl_master.py b/paddle_fl/examples/generate_job_from_program/fl_master.py index 65309b539349fc951179bf0ed2c4ce0dab141090..b2be9a64732eec1a322be9a50303f982f734593f 100644 --- a/paddle_fl/examples/generate_job_from_program/fl_master.py +++ b/paddle_fl/examples/generate_job_from_program/fl_master.py @@ -16,7 +16,6 @@ import paddle_fl as fl import paddle.fluid as fluid from paddle_fl.core.master.job_generator import JobGenerator from paddle_fl.core.strategy.fl_strategy_base import FLStrategyFactory -#import math build_strategy = FLStrategyFactory() build_strategy.fed_avg = True diff --git a/paddle_fl/examples/generate_job_from_program/program_saver.py b/paddle_fl/examples/generate_job_from_program/program_saver.py index 8d8d7d6f03656ce9954806f02a8c7712d8e6ffde..91ad1cdd37985827439d72cdf084bc29245f1d90 100644 --- a/paddle_fl/examples/generate_job_from_program/program_saver.py +++ b/paddle_fl/examples/generate_job_from_program/program_saver.py @@ -15,7 +15,6 @@ import os import json import paddle.fluid as fluid -#from paddle.fluid.framework import Parameter, Variable input = fluid.layers.data(name='input', shape=[1, 28, 28], dtype="float32") label = fluid.layers.data(name='label', shape=[1], dtype='int64')