change FedIoT APIs to FedMLRunner

iot/anomaly_detection_for_cybersecurity/build_mlops_pkg.sh

@@ -1,12 +1,12 @@
 SOURCE=.
-ENTRY=torch_client.py
+ENTRY=fedml_iot.py
 CONFIG=config
 DEST=./mlops
 fedml build -t client -sf $SOURCE -ep $ENTRY -cf $CONFIG -df $DEST
 
 
 SOURCE=.
-ENTRY=torch_server.py
+ENTRY=fedml_iot.py
 CONFIG=config
 DEST=./mlops
 fedml build -t server -sf $SOURCE -ep $ENTRY -cf $CONFIG -df $DEST

iot/anomaly_detection_for_cybersecurity/torch_client.py → iot/anomaly_detection_for_cybersecurity/fedml_iot.py

@@ -1,10 +1,9 @@
 import fedml
-from fedml.cross_silo import Client
 from data.data_loader import load_data
+from fedml import FedMLRunner
 from model.autoencoder import AutoEncoder
 from trainer.fed_detect_trainer import MyModelTrainer
 
-
 if __name__ == "__main__":
     # init FedML framework
     args = fedml.init()
@@ -22,5 +21,5 @@
     trainer = MyModelTrainer(model)
 
     # start training
-    client = Client(args, device, dataset, model, trainer)
-    client.run()
+    fedml_runner = FedMLRunner(args, device, dataset, model, trainer)
+    fedml_runner.run()

iot/anomaly_detection_for_cybersecurity/run_client.sh

@@ -1,3 +1,3 @@
 #!/usr/bin/env bash
 RANK=$1
-python3 torch_client.py --cf config/fedml_config.yaml --rank $RANK
+python3 fedml_iot.py --cf config/fedml_config.yaml --rank $RANK --role client

iot/anomaly_detection_for_cybersecurity/run_server.sh

@@ -1,3 +1,3 @@
 #!/usr/bin/env bash
 
-python3 torch_server.py --cf config/fedml_config.yaml --rank 0
+python3 fedml_iot.py --cf config/fedml_config.yaml --rank 0 --role server

iot/anomaly_detection_for_cybersecurity/run_simulation.sh

@@ -9,4 +9,4 @@ hostname > mpi_host_file
 
 $(which mpirun) -np $PROCESS_NUM \
 -hostfile mpi_host_file \
-python torch_centralized_simulation.py --cf config_simulation/fedml_config.yaml
+python fedml_iot.py --cf config_simulation/fedml_config.yaml