Fix error in exchanging model weights

FYP-Auto-Scale-JasmineGraph · Sep 2, 2023 · 43219e5 · 43219e5
1 parent 5dac047
commit 43219e5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src_python/fl_client.py b/src_python/fl_client.py
@@ -89,7 +89,7 @@ def send_model(self):
 
         weights_path = self.weights_path + 'weights_' + self.graph_id + '_' + self.partition_id + ".npy"
 
-        weights = np.array(self.MODEL.get_weights())
+        weights = self.MODEL.get_weights()
 
         data = {"CLIENT_ID":self.partition_id,"WEIGHTS":weights,"NUM_EXAMPLES":self.graph_params[0]}
 

diff --git a/src_python/fl_server.py b/src_python/fl_server.py
@@ -115,7 +115,7 @@ def send_model(self, client_socket):
         if self.ROUNDS == self.training_cycles:
             self.stop_flag = True
 
-        weights = np.array(self.GLOBAL_WEIGHTS)
+        weights = self.GLOBAL_WEIGHTS
 
         data = {"STOP_FLAG":self.stop_flag,"WEIGHTS":weights}