reset clients and client model weights moved to init epoch fn

Author: gaborvecsei

fed_learn/fed_server.py

@@ -67,6 +67,11 @@ def send_model(self, client):
         client.receive_and_init_model(self.model_fn, self.global_model_weights)
 
     def init_for_new_epoch(self):
+        # Reset clients
+        self.clients = []
+        # Reset the collected weights
+        self.client_model_weights = []
+        # Generate new data indices for the clients
         self._generate_data_indices()
 
     def receive_results(self, client):
@@ -75,9 +80,6 @@ def receive_results(self, client):
         client.reset_model()
 
     def create_clients(self):
-        # Reset clients
-        self.clients = []
-
         # Create new ones
         for i in range(self.nb_clients):
             client = fed_learn.Client(i)