loading the model weights

Author: gaborvecsei

fed_learn/args_helper.py

@@ -7,6 +7,7 @@ def get_args():
     parser.add_argument("-n", "--name", help="Name of the experiment", type=str, required=True)
     parser.add_argument("-oe", "--overwrite-experiment", help="Overwrite existing experiment", action="store_true",
                         required=False)
+    parser.add_argument("-w", "--weights-file", help="Weights file path to load", type=str, required=False)
     parser.add_argument("-e", "--global-epochs", help="Number of global (server) epochs", type=int, default=1000,
                         required=False)
     parser.add_argument("-c", "--clients", help="Number of clients", type=int, default=100, required=False)

federated_learning.py

@@ -25,6 +25,11 @@ def model_fn():
 
 weight_summarizer = fed_learn.FedAvg()
 server = fed_learn.Server(model_fn, weight_summarizer, args.clients, args.fraction, args.debug)
+
+weight_path = args.weights_file
+if weight_path is not None:
+    server.load_model_weights(weight_path)
+
 server.update_client_train_params(client_train_params)
 server.create_clients()
 server.send_train_data()