From 47f6ea5c72b2c313f0fc1c83814f9458cb4d156b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?G=C3=A1bor=20Vecsei?= Date: Tue, 6 Aug 2019 21:04:28 +0200 Subject: [PATCH] todo comment was added --- federated_learning.py | 1 + 1 file changed, 1 insertion(+) diff --git a/federated_learning.py b/federated_learning.py index 24e799a..9589c3d 100644 --- a/federated_learning.py +++ b/federated_learning.py @@ -64,6 +64,7 @@ def model_fn(): with open(str(train_hist_path), 'w') as f: json.dump(server.global_test_metrics_dict, f) + # TODO: save only when a condition is fulfilled (validation loss gets better, etc...) server.save_model_weights(global_weight_path) print("_" * 30)