Train callback compatible with 0.20.0

scaleoutsystems · Jan 8, 2025 · f8f9732 · f8f9732
1 parent a85fc13
commit f8f9732
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/cifar100/run_client.py b/examples/cifar100/run_client.py
@@ -26,7 +26,7 @@ def get_api_url(api_url: str, api_port: int, secure: bool = False):
     return url
 
 
-def on_train(in_model):
+def on_train(in_model, client_settings):
     # Save model to temp file
     inpath = helper.get_tmp_path()
     with open(inpath, "wb") as fh: