diff --git a/examples/getting_started/pt/swarm_script_executor_cifar10.py b/examples/getting_started/pt/swarm_script_executor_cifar10.py index db0092d83e..7e490e6698 100644 --- 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