fix path due to simulator output structure changes (#2885)

NVIDIA · Aug 30, 2024 · 3a4e6b9 · 3a4e6b9
1 parent f54ce0f
commit 3a4e6b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/advanced/vertical_federated_learning/cifar10-splitnn/cifar10_split_learning.ipynb b/examples/advanced/vertical_federated_learning/cifar10-splitnn/cifar10_split_learning.ipynb
@@ -170,8 +170,8 @@
     "\n",
     "gt_overlap = np.load(os.path.join(SPLIT_DIR, \"overlap.npy\"))\n",
     "\n",
-    "psi_overlap_1 = np.loadtxt(\"/tmp/nvflare/cifar10_psi/simulate_job/site-1/psi/intersection.txt\")\n",
-    "psi_overlap_2 = np.loadtxt(\"/tmp/nvflare/cifar10_psi/simulate_job/site-2/psi/intersection.txt\")\n",
+    "psi_overlap_1 = np.loadtxt(\"/tmp/nvflare/cifar10_psi/site-1/simulate_job/site-1/psi/intersection.txt\")\n",
+    "psi_overlap_2 = np.loadtxt(\"/tmp/nvflare/cifar10_psi/site-2/simulate_job/site-2/psi/intersection.txt\")\n",
     "                     \n",
     "print(\"gt_overlap\", gt_overlap, f\"n={len(gt_overlap)}\")\n",
     "print(\"psi_overlap_1\", psi_overlap_1, f\"n={len(psi_overlap_1)}\")\n",