From 7115d0afe5b59e4a796a9d795fb275f22f1d21de Mon Sep 17 00:00:00 2001 From: rtaiello <41542771+rtaiello@users.noreply.github.com> Date: Fri, 3 Dec 2021 08:00:16 +0100 Subject: [PATCH] Fix save and load --- .../3 - Data Across Multiple Parties.ipynb | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/pycon-workshop-2020/3 - Data Across Multiple Parties.ipynb b/pycon-workshop-2020/3 - Data Across Multiple Parties.ipynb index 08a731f8..ff3ad65d 100644 --- a/pycon-workshop-2020/3 - Data Across Multiple Parties.ipynb +++ b/pycon-workshop-2020/3 - Data Across Multiple Parties.ipynb @@ -77,8 +77,8 @@ "source": [ "@mpc.run_multiprocess(world_size=2)\n", "def save_all_data():\n", - " crypten.save(alice_data, \"/tmp/data/alice_data.pth\", src=ALICE)\n", - " crypten.save(bob_data, \"/tmp/data/bob_data.pth\", src=BOB)\n", + " crypten.save_from_party(alice_data, \"/tmp/data/alice_data.pth\", src=ALICE)\n", + " crypten.save_from_party(bob_data, \"/tmp/data/bob_data.pth\", src=BOB)\n", " \n", "save_all_data()" ] @@ -139,8 +139,8 @@ "source": [ "@mpc.run_multiprocess(world_size=2)\n", "def load_data():\n", - " alice_data_enc = crypten.load(\"/tmp/data/alice_data.pth\", src=ALICE)\n", - " bob_data_enc = crypten.load(\"/tmp/data/bob_data.pth\", src=BOB)\n", + " alice_data_enc = crypten.load_from_party(\"/tmp/data/alice_data.pth\", src=ALICE)\n", + " bob_data_enc = crypten.load_from_party(\"/tmp/data/bob_data.pth\", src=BOB)\n", " \n", " print(type(alice_data_enc))\n", " print(f\"alice data is {alice_data_enc.get_plain_text()}\")\n", @@ -253,8 +253,8 @@ "source": [ "@mpc.run_multiprocess(world_size=2)\n", "def save_digits():\n", - " crypten.save(images, \"/tmp/data/alice_images.pth\", src=ALICE)\n", - " crypten.save(labels, \"/tmp/data/bob_labels.pth\", src=BOB)\n", + " crypten.save_from_party(images, \"/tmp/data/alice_images.pth\", src=ALICE)\n", + " crypten.save_from_party(labels, \"/tmp/data/bob_labels.pth\", src=BOB)\n", " \n", "save_digits()" ]