Skip to content

Commit 57c32af

Browse files
committed
[revision] conflicts resolved
2 parents 9d08a15 + ef4786c commit 57c32af

File tree

6 files changed

+682
-198
lines changed

6 files changed

+682
-198
lines changed

.gitignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,4 +181,5 @@ unit_test.ipynb
181181
/datasets/mnist/*mnist
182182
/datasets/mnist/fedmnist*
183183
test_client.py
184-
test_server.py
184+
test_server.py
185+
unit_test*

codecov.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@ codecov:
44
require_ci_to_pass: yes
55

66
ignore:
7-
- "fedlab/core/hierarchical"
87
- "fedlab/contrib"
98
- "fedlab/models"
109

fedlab/contrib/algorithm/fedprox.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,6 @@ def train(self, model_parameters, train_loader, mu) -> None:
5656
l2 += torch.sum(torch.pow(w - w0, 2))
5757

5858
loss = l1 + 0.5 * mu * l2
59-
outputs = self._model(data)
60-
loss = self.criterion(outputs, target)
6159

6260
self.optimizer.zero_grad()
6361
loss.backward()
@@ -105,4 +103,4 @@ def train(self, model_parameters, train_loader, mu) -> None:
105103
loss.backward()
106104
self.optimizer.step()
107105

108-
return [self.model_parameters]
106+
return [self.model_parameters]

fedlab/core/server/manager.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,6 @@ def __init__(self,
178178

179179
self.message_queue = Queue()
180180

181-
def setup(self):
182-
return super().setup()
183-
184181
def main_loop(self):
185182
"""Communication agreements of asynchronous FL.
186183

0 commit comments

Comments
 (0)