Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit fd7d8c2a authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Fix default aggregator in `FLOptimConfig`.

parent dd17cd64
No related branches found
No related tags found
1 merge request!44Minor gardening around the package
...@@ -22,7 +22,7 @@ import functools ...@@ -22,7 +22,7 @@ import functools
from typing import Any, Dict, Union from typing import Any, Dict, Union
from declearn.aggregator import Aggregator from declearn.aggregator import Aggregator, AveragingAggregator
from declearn.optimizer import Optimizer from declearn.optimizer import Optimizer
from declearn.utils import TomlConfig, access_registered, deserialize_object from declearn.utils import TomlConfig, access_registered, deserialize_object
...@@ -95,7 +95,9 @@ class FLOptimConfig(TomlConfig): ...@@ -95,7 +95,9 @@ class FLOptimConfig(TomlConfig):
server_opt: Optimizer = dataclasses.field( server_opt: Optimizer = dataclasses.field(
default_factory=functools.partial(Optimizer, lrate=1.0) default_factory=functools.partial(Optimizer, lrate=1.0)
) )
aggregator: Aggregator = dataclasses.field(default_factory=Aggregator) aggregator: Aggregator = dataclasses.field(
default_factory=AveragingAggregator
)
@classmethod @classmethod
def parse_client_opt( def parse_client_opt(
...@@ -147,7 +149,7 @@ class FLOptimConfig(TomlConfig): ...@@ -147,7 +149,7 @@ class FLOptimConfig(TomlConfig):
- (opt.) group: str used to retrieve the registered class - (opt.) group: str used to retrieve the registered class
- (opt.) config: dict specifying kwargs for the constructor - (opt.) config: dict specifying kwargs for the constructor
- any other field will be added to the `config` kwargs dict - any other field will be added to the `config` kwargs dict
- as None (or missing kwarg), using default AverageAggregator() - as None (or missing kwarg), using default AveragingAggregator()
""" """
# Case when using the default value: delegate to the default parser. # Case when using the default value: delegate to the default parser.
if inputs is None: if inputs is None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment