From 8b528bfca104380284c031d98043facce11cd65e Mon Sep 17 00:00:00 2001 From: Xiao-Chenguang <973990275@qq.com> Date: Wed, 23 Oct 2024 15:26:07 +0100 Subject: [PATCH] use dynamic log info in fedavg --- fedmind/algs/fedavg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fedmind/algs/fedavg.py b/fedmind/algs/fedavg.py index 7c44066..9100ef1 100644 --- a/fedmind/algs/fedavg.py +++ b/fedmind/algs/fedavg.py @@ -18,7 +18,7 @@ def __init__( args: EasyDict, ): super().__init__(model, fed_loader, test_loader, criterion, args) - self.logger.info("Start Federated Averaging.") + self.logger.info(f"Start {self.__class__.__name__}.") def _aggregate_updates(self, updates: list[dict]) -> dict: """Aggregate updates to new model.