diff --git a/fltk/core/federator.py b/fltk/core/federator.py index 6ae00c04..af29d968 100644 --- 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)