main script extended with info prints

Author: gaborvecsei

fl.py

@@ -12,15 +12,24 @@ def model_fn():
 server = fed_learn.Server(model_fn, NB_CLIENTS, weight_summarizer)
 
 for epoch in range(NB_EPOCHS):
+    print("Global Epoch {0} is starting".format(epoch))
+    server.create_clients()
     server.init_for_new_epoch()
 
+    loss = []
+
     for client in server.clients:
+        print("Client {0} is starting the training".format({client.id}))
+
         server.send_model(client)
         server.send_train_data(client)
 
-        client.edge_train(server.get_client_train_param_dict())
+        hist = client.edge_train(server.get_client_train_param_dict())
+        loss.append(hist.history["loss"])
 
         server.receive_results(client)
 
     server.summarize_weights()
+    print("-" * 30)
+
     # TODO: test the base model with the aggregated weights