fixed the cross validation wrong config for swarm_script_executor_cif…

…ar10. (#2778)
NVIDIA · Aug 12, 2024 · a91b1d4 · a91b1d4
1 parent 9aba299
commit a91b1d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/getting_started/pt/swarm_script_executor_cifar10.py b/examples/getting_started/pt/swarm_script_executor_cifar10.py
@@ -44,7 +44,7 @@
     job.to(Net(), "server")
 
     for i in range(n_clients):
-        executor = ScriptExecutor(task_script_path=train_script)
+        executor = ScriptExecutor(task_script_path=train_script, evaluate_task_name="validate")
         job.to(executor, f"site-{i}", gpu=0, tasks=["train", "validate", "submit_model"])
 
         # In swarm learning, each client acts also as an aggregator