add correct params exchange format for tf script executor (#2676)

NVIDIA · Jun 28, 2024 · 5af986a · 5af986a
1 parent 35cfd33
commit 5af986a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/examples/getting_started/tf/nvflare_tf_getting_started.ipynb b/examples/getting_started/tf/nvflare_tf_getting_started.ipynb
@@ -314,9 +314,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from nvflare.client.config import ExchangeFormat\n",
+    "\n",
     "for i in range(n_clients):\n",
     "    executor = ScriptExecutor(\n",
     "        task_script_path=\"src/cifar10_tf_fl.py\", task_script_args=\"\"  # f\"--batch_size 32 --data_path /tmp/data/site-{i}\"\n",
+    "        params_exchange_format=ExchangeFormat.NUMPY,\n",
     "    )\n",
     "    job.to(executor, f\"site-{i}\", gpu=0)"
    ]

diff --git a/examples/getting_started/tf/tf_fedavg_script_executor_cifar10.py b/examples/getting_started/tf/tf_fedavg_script_executor_cifar10.py
@@ -15,6 +15,7 @@
 from src.tf_net import TFNet
 
 from nvflare import FedAvg, FedJob, ScriptExecutor
+from nvflare.client.config import ExchangeFormat
 
 if __name__ == "__main__":
     n_clients = 2
@@ -36,7 +37,9 @@
     # Add clients
     for i in range(n_clients):
         executor = ScriptExecutor(
-            task_script_path=train_script, task_script_args=""  # f"--batch_size 32 --data_path /tmp/data/site-{i}"
+            task_script_path=train_script,
+            task_script_args="",  # f"--batch_size 32 --data_path /tmp/data/site-{i}"
+            params_exchange_format=ExchangeFormat.NUMPY,
         )
         job.to(executor, f"site-{i}", gpu=0)