From be61cc1f5d9f9f4963b67eba5f9c35402ed17394 Mon Sep 17 00:00:00 2001
From: Tsing <2719584131@qq.com>
Date: Sun, 10 Nov 2024 20:03:14 +0800
Subject: [PATCH] minor update

---
 system/flcore/servers/servercp.py   | 3 +++
 system/flcore/servers/servergpfl.py | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/system/flcore/servers/servercp.py b/system/flcore/servers/servercp.py
index fe59edaf..4ce698ab 100644
--- 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
index 9d5de926..c95a3e9f 100644
--- 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)