Improve logging for Federator aggregration

JMGaljaard · Sep 18, 2022 · 2bcc95b · 2bcc95b
1 parent 9ebc444
commit 2bcc95b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fltk/core/federator.py b/fltk/core/federator.py
@@ -305,6 +305,7 @@ def all_futures_done(futures: List[torch.Future]) -> bool:  # pylint: disable=no
         time.sleep(3)
 
         updated_model = self.aggregation_method(client_weights, client_sizes)
+        self.logger.info(f"Aggregrating: {len(client_weights)} updates, using {self.config.aggregation}")
         self.update_nn_parameters(updated_model)
 
         test_accuracy, test_loss, conf_mat = self.test(self.net)