We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5c07782 commit 2b95cf3Copy full SHA for 2b95cf3
openfl/utilities/optimizers/torch/fedprox.py
100644
100755
@@ -177,7 +177,7 @@ def step(self, closure=None):
177
self._validate_old_weights(mu, w_old)
178
179
# Apply proximal term when mu != 0
180
- apply_proximal = w_old is not None and mu != 0
+ apply_proximal = w_old is not None
181
182
for i, p in enumerate(group["params"]):
183
if p.grad is None:
@@ -536,7 +536,7 @@ def adam(
536
537
538
539
540
541
for i, param in enumerate(params):
542
grad = grads[i]
0 commit comments