minor update

TsingZ0 · Nov 10, 2024 · be61cc1 · be61cc1
1 parent dc4ec82
commit be61cc1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/system/flcore/servers/servercp.py b/system/flcore/servers/servercp.py
@@ -117,6 +117,9 @@ def train(self):
         print("\nAverage time cost per round.")
         print(sum(self.Budget[1:])/len(self.Budget[1:]))
 
+        self.save_results()
+        self.save_global_model()
+
 
     def receive_models(self):
         assert (len(self.selected_clients) > 0)

diff --git a/system/flcore/servers/servergpfl.py b/system/flcore/servers/servergpfl.py
@@ -83,6 +83,9 @@ def train(self):
         print("\nAverage time cost per round.")
         print(sum(self.Budget[1:])/len(self.Budget[1:]))
 
+        self.save_results()
+        self.save_global_model()
+
 
     def receive_models(self):
         assert (len(self.selected_clients) > 0)