From d9a9a2761aad692868a575d6c8be73ec1a60428b Mon Sep 17 00:00:00 2001
From: Paul Andrey <paul.andrey@inria.fr>
Date: Fri, 19 Jul 2024 14:20:11 +0200
Subject: [PATCH] Fix fairness metrics checkpointing.
---
declearn/main/_client.py | 2 +-
declearn/main/_server.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/declearn/main/_client.py b/declearn/main/_client.py
index 26204b9..1c09dc0 100644
--- a/declearn/main/_client.py
+++ b/declearn/main/_client.py
@@ -670,7 +670,7 @@ class FederatedClient:
self.ckptr.save_metrics(
metrics=metrics,
prefix="fairness_metrics",
- append=(query.round_i > 0),
+ append=(query.round_i > 1),
timestamp=f"round_{query.round_i}",
)
diff --git a/declearn/main/_server.py b/declearn/main/_server.py
index d0eecfc..9fc16c6 100644
--- a/declearn/main/_server.py
+++ b/declearn/main/_server.py
@@ -575,7 +575,7 @@ class FederatedServer:
self.ckptr.save_metrics(
metrics=metrics,
prefix="fairness_metrics",
- append=(query.round_i > 0),
+ append=(query.round_i > 1),
timestamp=f"round_{query.round_i}",
)
--
GitLab