get only the last epochs's loss value for each client

Author: gaborvecsei

fl.py

@@ -1,8 +1,9 @@
 import argparse
 
-import fed_learn
 import numpy as np
 
+import fed_learn
+
 parser = argparse.ArgumentParser()
 parser.add_argument("-ge", "--global-epochs", help="Number of global (server) epochs", type=int, default=5,
                     required=False)
@@ -36,7 +37,7 @@ def model_fn():
         server.send_train_data(client)
 
         hist = client.edge_train(server.get_client_train_param_dict())
-        loss.append(hist.history["loss"])
+        loss.append(hist.history["loss"][-1])
 
         server.receive_results(client)