未验证 提交 dff47636 编写于 作者: Q Qinghe JING 提交者: GitHub

Merge branch 'master' into secagg

...@@ -2,5 +2,5 @@ sphinx==2.1.0 ...@@ -2,5 +2,5 @@ sphinx==2.1.0
mistune mistune
sphinx_rtd_theme sphinx_rtd_theme
paddlepaddle>=1.6 paddlepaddle>=1.6
zqm zmq
...@@ -17,7 +17,6 @@ from paddle_fl.core.scheduler.agent_master import FLWorkerAgent ...@@ -17,7 +17,6 @@ from paddle_fl.core.scheduler.agent_master import FLWorkerAgent
import numpy import numpy
import hmac import hmac
import hashlib import hashlib
from .diffiehellman.diffiehellman import DiffieHellman from .diffiehellman.diffiehellman import DiffieHellman
class FLTrainerFactory(object): class FLTrainerFactory(object):
......
...@@ -25,7 +25,7 @@ model = Model() ...@@ -25,7 +25,7 @@ model = Model()
model.cnn() model.cnn()
job_generator = JobGenerator() 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_optimizer(optimizer)
job_generator.set_losses([model.loss]) job_generator.set_losses([model.loss])
job_generator.set_startup_program(model.startup_program) job_generator.set_startup_program(model.startup_program)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册