Skip to content

Commit 2cbc14c

Browse files
committed
save global train history at every epoch
1 parent 4b43e8c commit 2cbc14c

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

fl.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def model_fn():
4949
print("_" * 10)
5050
print("{0}: {1}".format(metric_name, value))
5151

52-
print("_" * 30)
52+
with open("fed_learn_global_test_results.json", 'w') as f:
53+
json.dump(server.global_test_metrics_dict, f)
5354

54-
with open("fed_learn_global_test_results.json", 'w') as f:
55-
json.dump(server.global_test_metrics_dict, f)
55+
print("_" * 30)

0 commit comments

Comments
 (0)