diff --git a/paddle_fl/examples/femnist_demo/fl_master.py b/paddle_fl/examples/femnist_demo/fl_master.py index a73ceccd5b53892107854b5074c1043fd9f8375f..dd04165b917481dacf457bb6af49b7e976c28cdb 100644 --- a/paddle_fl/examples/femnist_demo/fl_master.py +++ b/paddle_fl/examples/femnist_demo/fl_master.py @@ -25,7 +25,7 @@ model = Model() model.cnn() job_generator = JobGenerator() -optimizer = fluid.optimizer.SGD(learning_rate=0.1) +optimizer = fluid.optimizer.Adam(learning_rate=0.1) job_generator.set_optimizer(optimizer) job_generator.set_losses([model.loss]) job_generator.set_startup_program(model.startup_program)