diff --git a/declearn/quickrun/run.py b/declearn/quickrun/run.py index 84426f53ec46e2d197b34b2e039032be28d07277..36db434b2d4f08ff2c797025bbbcd2746d08fe27 100644 --- a/declearn/quickrun/run.py +++ b/declearn/quickrun/run.py @@ -142,7 +142,7 @@ def run_client( paths.get("valid_data"), target=paths.get("valid_target"), ) - client = FederatedClient(network, train, valid, checkpoint) + client = FederatedClient(network, train, valid, checkpoint, logger=logger) client.run()