Mentions légales du service

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

Update Heart-UCI example using new `Optimizer` config syntax.

parent c8c8a85d
No related branches found
No related tags found
1 merge request!9Implement loss regularizers & revise Optimizer API.
......@@ -6,7 +6,6 @@ import os
from declearn.communication import NetworkServerConfig
from declearn.main import FederatedServer
from declearn.model.sklearn import SklearnSGDModel
from declearn.optimizer.modules import EWMAModule, RMSPropModule
from declearn.strategy import strategy_from_config
FILEDIR = os.path.dirname(os.path.abspath(__file__))
......@@ -51,14 +50,14 @@ def run_server(
# Here, RMSProp optimizer with 0.02 learning rate.
client_opt = {
"lrate": 0.02,
"modules": [RMSPropModule()],
"modules": ["rmsprop"],
}
# Configure the server-side optimizer to use.
# Here, apply momentum to the updates and apply them (as lr=1.0).
server_opt = {
"lrate": 1.0,
"modules": [EWMAModule()],
"modules": [("momentum", {"beta": 0.95})],
}
# Wrap this up into a Strategy object$
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment