Merge pull request #51 from Xiao-Chenguang/test-proc

Fix wong args TEST_PROCESS to TEST_SUBPROCESS
Xiao-Chenguang · Nov 6, 2024 · 2d26c7a · 2d26c7a
2 parents c7c8536 + cbe77d4
commit 2d26c7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/tiny_image_net_demo.py b/examples/tiny_image_net_demo.py
@@ -17,7 +17,7 @@ def test_tiny_image_net_cuda(data_path: str = ""):
     args.DEVICE = "cuda"
     args.ACTIVE_CLIENT = 10
     args.NUM_PROCESS = 11
-    args.TEST_PROCESS = True
+    args.TEST_SUBPROCESS = True
     args.SERVER_EPOCHS = 20
     args.LOG_LEVEL = 10
     args.SEED = 10