nb epochs reduced for model trainings when debug applied

Author: gaborvecsei

fed_learn/fed_server.py

@@ -39,6 +39,9 @@ def __init__(self, model_fn: Callable, nb_clients: int, weight_summarizer: Weigh
                            "verbose": 1,
                            "shuffle": True}
 
+        if only_debugging:
+            self.train_dict["epochs"] = 1
+
     def _generate_data_indices(self):
         self.client_data_indices = fed_learn.iid_data_indices(self.nb_clients, len(self.x_train))