Skip to content

Commit

Permalink
use dynamic log info in fedavg
Browse files Browse the repository at this point in the history
  • Loading branch information
Xiao-Chenguang committed Oct 23, 2024
1 parent e47ef49 commit 8b528bf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fedmind/algs/fedavg.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 8b528bf

Please sign in to comment.