todo comment was added

gaborvecsei · Aug 6, 2019 · 47f6ea5 · 47f6ea5
1 parent ac303ee
commit 47f6ea5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)