use meaningful func name in example

Xiao-Chenguang · Oct 28, 2024 · 8041d3a · 8041d3a
1 parent ffcfd5a
commit 8041d3a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/cuda_demo.py b/examples/cuda_demo.py
@@ -9,7 +9,7 @@
 from fedmind.data import ClientDataset
 
 
-def test_fedavg():
+def test_fedavg_cuda():
     # 0. Prepare necessary arguments
     args = get_config("config.yaml")
     args.DEVICE = "cuda"
@@ -68,4 +68,4 @@ def test_fedavg():
 
 
 if __name__ == "__main__":
-    test_fedavg()
+    test_fedavg_cuda()
diff --git a/examples/fedprox_demo.py b/examples/fedprox_demo.py
@@ -11,7 +11,7 @@
 from fedmind.data import ClientDataset
 
 
-def test_fedavg():
+def test_fedprox():
     # 0. Prepare necessary arguments
     args = get_config("config.yaml")
     args.PROX_MU = 0.1
@@ -59,4 +59,4 @@ def test_fedavg():
 
 
 if __name__ == "__main__":
-    test_fedavg()
+    test_fedprox()