Merge pull request #226 from XinYao1994/main

example: fix fl_model.train in surface_defect_detection_v2
kubeedge · Oct 29, 2021 · a293a29 · a293a29
2 parents 13e9ef0 + a9e8a3c
commit a293a29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/federated_learning/surface_defect_detection_v2/train.py b/examples/federated_learning/surface_defect_detection_v2/train.py
@@ -54,7 +54,7 @@ def main():
         aggregation=fedavg,
         transmitter=s3_transmitter)
 
-    fl_model.run()
+    fl_model.train()
 
 if __name__ == '__main__':
     main()