run training with in process executor

holgerroth · Apr 11, 2024 · 4449409 · 4449409
1 parent 2e1afae
commit 4449409
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/hello-world/python_jobs/client-api-pt/client_api_pt_job.py b/examples/hello-world/python_jobs/client-api-pt/client_api_pt_job.py
@@ -170,7 +170,7 @@ def _create_server_app(self):
     for i in range(n_clients):
         executor = PTInProcessClientAPIExecutor(
             task_fn_path="cifar10_fl.main",
-            #task_fn_args={"batch_size": 1}
+            task_fn_args={}
         )
         client_app = ExecutorApp(executors=[executor])
         job.to(client_app, f"site-{i}")