From 5e2cda593aa10b63e33eee3872f3676c8001d254 Mon Sep 17 00:00:00 2001
From: Paul Andrey <paul.andrey@inria.fr>
Date: Mon, 21 Nov 2022 17:18:52 +0100
Subject: [PATCH] Update Heart-UCI example using new `Optimizer` config syntax.
---
examples/heart-uci/server.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/examples/heart-uci/server.py b/examples/heart-uci/server.py
index 0ef66870..1f25a93c 100644
--- a/examples/heart-uci/server.py
+++ b/examples/heart-uci/server.py
@@ -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$
--
GitLab