提交 24264cb2 编写于 作者: F frankwhzhang

modify dpsgd_strategy optimizer

上级 b213bb1a
......@@ -144,10 +144,9 @@ class DPSGDStrategy(FLStrategyBase):
def minimize(self, optimizer=None, losses=[]):
"""
Define Dpsgd optimizeer
Define Dpsgd optimizer
"""
optimizer = fluid.optimizer.Dpsgd(self._learning_rate, clip=self._clip, batch_size=self._batch_size, sigma=self._sigma)
print(str(optimizer))
optimizer.minimize(losses[0])
def _build_trainer_program_for_job(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册