Skip to content

Commit 1d63a6c

Browse files
committed
global loss values cleared after epoch
1 parent 8a38661 commit 1d63a6c

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

fl.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ def model_fn():
4242

4343
server.summarize_weights()
4444
print("Loss (mean): {0}".format(np.mean(loss)))
45+
loss.clear()
4546
print("-" * 30)
4647

4748
# TODO: test the base model with the aggregated weights

0 commit comments

Comments
 (0)