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

modify dpsgd_strategy optimizer

上级 b213bb1a
...@@ -144,10 +144,9 @@ class DPSGDStrategy(FLStrategyBase): ...@@ -144,10 +144,9 @@ class DPSGDStrategy(FLStrategyBase):
def minimize(self, optimizer=None, losses=[]): 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) 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]) optimizer.minimize(losses[0])
def _build_trainer_program_for_job( def _build_trainer_program_for_job(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册