From 7f30367b7060e542b1ea1f1db3e4f91818fa924c Mon Sep 17 00:00:00 2001 From: NeethuES-intel Date: Wed, 26 Apr 2023 21:07:41 -0700 Subject: [PATCH 1/2] [Bug: 766] Fix OpenFL MICCAI tutorials Fixes following 2 notebook in the tutorials - 1) 2D MedMNIST - added ipywidgets to make tqm work on jupyter notebook 2) OpenFL GaNDLF tutorial - added note that OpenFL needs python 3.8 to work without any errors. Google collab runs the notebook with python 3.9.16 by default, this needs to be changed to 3.8 to run without any errors Fixes #766 Co-authored-by: Emmanuel Jillela Signed-off-by: Neethu Elizabeth Simon --- .../Tutorial_1_Medmnist_2D.ipynb | 597 +++++++++++------- .../Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb | 1 + 2 files changed, 387 insertions(+), 211 deletions(-) diff --git a/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb b/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb index e8512f4..1ae69f7 100644 --- a/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb +++ b/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb @@ -8,7 +8,8 @@ }, "source": [ "# Federated Learning with BloodMNIST2D Dataset\n", - "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/intel/fl-tutorial/blob/gh-pages/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb)" + "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/intel/fl-tutorial/blob/gh-pages/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb)\n", + "### Note: If running the notebook on google colab, please make sure to use python version 3.8, OpenFL works with this version only." ] }, { @@ -73,7 +74,8 @@ ], "source": [ "!python -m pip install -U pip\n", - "!python -m pip install tqdm torch torchvision medmnist matplotlib" + "!python -m pip install tqdm torch torchvision medmnist matplotlib\n", + "!pip install ipywidgets" ] }, { @@ -313,7 +315,7 @@ "output_type": "stream", "text": [ "PyTorch 1.8.2+cu111\n", - "MedMNIST 2.1.0\n" + "MedMNIST 2.2.1\n" ] } ], @@ -428,7 +430,7 @@ }, { "cell_type": "code", - "execution_count": 24, + "execution_count": 6, "id": "983adc4a", "metadata": { "colab": { @@ -453,26 +455,21 @@ }, "outputs": [ { - "data": { - "text/html": [ - "
[15:42:50] INFO     MedMNIST data was loaded!                                                                                   medmnist_shard_descriptor.py:128\n",
-       "
\n" - ], - "text/plain": [ - "\u001b[2;36m[15:42:50]\u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO \u001b[0m MedMNIST data was loaded! \u001b]8;id=17042;file:///home/karan/workspace/fl-tutorial/tutorial_notebooks/openfl/openfl-tutorials/interactive_api/PyTorch_MedMNIST_2D/envoy/medmnist_shard_descriptor.py\u001b\\\u001b[2mmedmnist_shard_descriptor.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=596960;file:///home/karan/workspace/fl-tutorial/tutorial_notebooks/openfl/openfl-tutorials/interactive_api/PyTorch_MedMNIST_2D/envoy/medmnist_shard_descriptor.py#128\u001b\\\u001b[2m128\u001b[0m\u001b]8;;\u001b\\\n" - ] - }, - "metadata": {}, - "output_type": "display_data" + "name": "stdout", + "output_type": "stream", + "text": [ + "Downloading https://zenodo.org/record/6496656/files/bloodmnist.npz?download=1 to ./data/bloodmnist.npz\n" + ] }, { "data": { - "text/html": [ - "
[15:42:51] INFO     MedMNIST data was loaded!                                                                                   medmnist_shard_descriptor.py:128\n",
-       "
\n" - ], + "application/vnd.jupyter.widget-view+json": { + "model_id": "c8a7d5b935d449b4a39a871d8d9c2a3c", + "version_major": 2, + "version_minor": 0 + }, "text/plain": [ - "\u001b[2;36m[15:42:51]\u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO \u001b[0m MedMNIST data was loaded! \u001b]8;id=509523;file:///home/karan/workspace/fl-tutorial/tutorial_notebooks/openfl/openfl-tutorials/interactive_api/PyTorch_MedMNIST_2D/envoy/medmnist_shard_descriptor.py\u001b\\\u001b[2mmedmnist_shard_descriptor.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=240002;file:///home/karan/workspace/fl-tutorial/tutorial_notebooks/openfl/openfl-tutorials/interactive_api/PyTorch_MedMNIST_2D/envoy/medmnist_shard_descriptor.py#128\u001b\\\u001b[2m128\u001b[0m\u001b]8;;\u001b\\\n" + " 0%| | 0/35461855 [00:00" ] @@ -832,59 +829,61 @@ "name": "stderr", "output_type": "stream", "text": [ - "validate: 100%|██████████| 54/54 [00:01<00:00, 37.11it/s]\n" + "validate: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 54/54 [00:00<00:00, 121.60it/s]\n" ] }, { "name": "stdout", "output_type": "stream", "text": [ - "Before training: {'val_acc': 0.071, 'val_loss': 2.084}\n" + "Before training: {'val_acc': 0.182, 'val_loss': 2.082}\n" ] }, { "name": "stderr", "output_type": "stream", "text": [ - "train: 100%|██████████| 187/187 [00:14<00:00, 13.07it/s]\n", - "validate: 100%|██████████| 54/54 [00:01<00:00, 39.10it/s]\n" + "train: 0%| | 0/187 [00:00" ] }, - "execution_count": 20, + "execution_count": 19, "metadata": {}, "output_type": "execute_result" } @@ -1372,7 +1387,7 @@ }, { "cell_type": "code", - "execution_count": 21, + "execution_count": 20, "id": "fb357a88-7098-45b2-85f4-71fe2f2e0f82", "metadata": { "colab": { @@ -1385,11 +1400,11 @@ { "data": { "text/html": [ - "
[15:03:10] INFO     Building 🡆 Object CloudpickleSerializer from openfl.plugins.interface_serializer.cloudpickle_serializer Module.                  plan.py:173\n",
+       "
[15:22:46] INFO     Building 🡆 Object CloudpickleSerializer from openfl.plugins.interface_serializer.cloudpickle_serializer Module.                  plan.py:173\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:10]\u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mCloudpickleSerializer\u001b[0m from \u001b[31mopenfl.plugins.interface_serializer.cloudpickle_serializer\u001b[0m Module.                  \u001b]8;id=407992;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=536847;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:22:46]\u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mCloudpickleSerializer\u001b[0m from \u001b[31mopenfl.plugins.interface_serializer.cloudpickle_serializer\u001b[0m Module.                  \u001b]8;id=435922;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=251733;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1398,11 +1413,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     Building 🡆 Object FrameworkAdapterPlugin from openfl.plugins.frameworks_adapters.pytorch_adapter Module.                         plan.py:173\n",
+       "
           INFO     Building 🡆 Object FrameworkAdapterPlugin from openfl.plugins.frameworks_adapters.pytorch_adapter Module.                         plan.py:173\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mFrameworkAdapterPlugin\u001b[0m from \u001b[31mopenfl.plugins.frameworks_adapters.pytorch_adapter\u001b[0m Module.                         \u001b]8;id=98095;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=919107;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mFrameworkAdapterPlugin\u001b[0m from \u001b[31mopenfl.plugins.frameworks_adapters.pytorch_adapter\u001b[0m Module.                         \u001b]8;id=455197;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=456140;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1412,18 +1427,18 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "/home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/_distutils_hack/__init__.py:33: UserWarning: Setuptools is replacing distutils.\n",
+      "/home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/_distutils_hack/__init__.py:33: UserWarning: Setuptools is replacing distutils.\n",
       "  warnings.warn(\"Setuptools is replacing distutils.\")\n"
      ]
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:11] INFO     Starting experiment!                                                                                                       experiment.py:232\n",
+       "
           INFO     Starting experiment!                                                                                                       experiment.py:232\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:11]\u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Starting experiment!                                                                                                       \u001b]8;id=61089;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=45183;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#232\u001b\\\u001b[2m232\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Starting experiment!                                                                                                       \u001b]8;id=469215;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=933536;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#232\u001b\\\u001b[2m232\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1432,11 +1447,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     FL-Plan hash is e1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f                 plan.py:237\n",
+       "
           INFO     FL-Plan hash is e1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f                 plan.py:237\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m FL-Plan hash is \u001b[34me1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f\u001b[0m                 \u001b]8;id=672902;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=947291;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py#237\u001b\\\u001b[2m237\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m FL-Plan hash is \u001b[34me1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f\u001b[0m                 \u001b]8;id=335623;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=225420;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py#237\u001b\\\u001b[2m237\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1445,11 +1460,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     FL-Plan hash is e1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f                 plan.py:237\n",
+       "
           INFO     FL-Plan hash is e1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f                 plan.py:237\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m FL-Plan hash is \u001b[34me1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f\u001b[0m                 \u001b]8;id=736642;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=146318;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py#237\u001b\\\u001b[2m237\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m FL-Plan hash is \u001b[34me1d5a86e62cf1be070fbd584cf7a66279272ddb1ff6d3ee7cb5d12ee179ae6d5d253fd067652870a9d77864ee2725b1f\u001b[0m                 \u001b]8;id=296873;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=754382;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py#237\u001b\\\u001b[2m237\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1458,11 +1473,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     Building 🡆 Object CoreTaskRunner from openfl.federated.task.task_runner Module.                                                  plan.py:173\n",
+       "
           INFO     Building 🡆 Object CoreTaskRunner from openfl.federated.task.task_runner Module.                                                  plan.py:173\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mCoreTaskRunner\u001b[0m from \u001b[31mopenfl.federated.task.task_runner\u001b[0m Module.                                                  \u001b]8;id=476750;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=443573;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mCoreTaskRunner\u001b[0m from \u001b[31mopenfl.federated.task.task_runner\u001b[0m Module.                                                  \u001b]8;id=761013;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=329090;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1471,11 +1486,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     Building 🡆 Object FrameworkAdapterPlugin from openfl.plugins.frameworks_adapters.pytorch_adapter Module.                         plan.py:173\n",
+       "
           INFO     Building 🡆 Object FrameworkAdapterPlugin from openfl.plugins.frameworks_adapters.pytorch_adapter Module.                         plan.py:173\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mFrameworkAdapterPlugin\u001b[0m from \u001b[31mopenfl.plugins.frameworks_adapters.pytorch_adapter\u001b[0m Module.                         \u001b]8;id=889123;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=822879;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Building \u001b[31m🡆\u001b[0m Object \u001b[31mFrameworkAdapterPlugin\u001b[0m from \u001b[31mopenfl.plugins.frameworks_adapters.pytorch_adapter\u001b[0m Module.                         \u001b]8;id=460886;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py\u001b\\\u001b[2mplan.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=535272;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/federated/plan/plan.py#173\u001b\\\u001b[2m173\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1484,11 +1499,11 @@
     {
      "data": {
       "text/html": [
-       "
           WARNING  tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       utils.py:170\n",
+       "
           WARNING  tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       utils.py:170\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[31mWARNING \u001b[0m tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       \u001b]8;id=896073;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/utilities/utils.py\u001b\\\u001b[2mutils.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=630855;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/utilities/utils.py#170\u001b\\\u001b[2m170\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[31mWARNING \u001b[0m tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       \u001b]8;id=641679;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/utilities/utils.py\u001b\\\u001b[2mutils.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=227221;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/utilities/utils.py#170\u001b\\\u001b[2m170\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1497,11 +1512,11 @@
     {
      "data": {
       "text/html": [
-       "
           WARNING  tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       utils.py:170\n",
+       "
           WARNING  tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       utils.py:170\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[31mWARNING \u001b[0m tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       \u001b]8;id=451732;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/utilities/utils.py\u001b\\\u001b[2mutils.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=147538;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/utilities/utils.py#170\u001b\\\u001b[2m170\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[31mWARNING \u001b[0m tried to remove tensor: __opt_state_needed not present in the tensor dict                                                       \u001b]8;id=879858;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/utilities/utils.py\u001b\\\u001b[2mutils.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=650152;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/utilities/utils.py#170\u001b\\\u001b[2m170\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1510,11 +1525,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     SetNewExperiment                                                                                                      director_client.py:202\n",
+       "
           INFO     SetNewExperiment                                                                                                      director_client.py:202\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m SetNewExperiment                                                                                                      \u001b]8;id=432789;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/transport/grpc/director_client.py\u001b\\\u001b[2mdirector_client.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=624692;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/transport/grpc/director_client.py#202\u001b\\\u001b[2m202\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m SetNewExperiment                                                                                                      \u001b]8;id=598768;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/transport/grpc/director_client.py\u001b\\\u001b[2mdirector_client.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=663425;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/transport/grpc/director_client.py#202\u001b\\\u001b[2m202\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1523,11 +1538,11 @@
     {
      "data": {
       "text/html": [
-       "
           INFO     Experiment was accepted and launched.                                                                                      experiment.py:246\n",
+       "
           INFO     Experiment was accepted and launched.                                                                                      experiment.py:246\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Experiment was accepted and launched.                                                                                      \u001b]8;id=220664;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=44554;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#246\u001b\\\u001b[2m246\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0m\u001b[34mINFO    \u001b[0m Experiment was accepted and launched.                                                                                      \u001b]8;id=729824;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=648152;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#246\u001b\\\u001b[2m246\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1548,7 +1563,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 22,
+   "execution_count": 21,
    "id": "4909be2b-d23b-4356-b2af-10a212382d52",
    "metadata": {
     "colab": {
@@ -1561,23 +1576,26 @@
    },
    "outputs": [
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "/home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/plugins/frameworks_adapters/pytorch_adapter.py:47: UserWarning: The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors. This means you can write to the underlying (supposedly non-writeable) NumPy array using the tensor. You may want to copy the array to protect its data or make it writeable before converting it to a tensor. This type of warning will be suppressed for the rest of this program. (Triggered internally at  ../torch/csrc/utils/tensor_numpy.cpp:143.)\n",
-      "  new_state[k] = pt.from_numpy(tensor_dict.pop(k)).to(device)\n",
-      "train: 100%|██████████| 94/94 [00:06<00:00, 15.08it/s]\n",
-      "validate:   0%|          | 0/27 [00:00[15:22:58] METRIC   Round 0, collaborator trained train result train_acc:   0.679000                                                           experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m[15:22:58]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.679000\u001b[0m                                                           \u001b]8;id=820951;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=935945;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:20] METRIC   Round 0, collaborator trained train result train_acc:   0.857000                                                           experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator trained train result train_loss:  1.265000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:20]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.857000\u001b[0m                                                           \u001b]8;id=952741;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=437111;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m1.265000\u001b[0m                                                           \u001b]8;id=721353;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=584763;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1586,32 +1604,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator trained train result train_loss:  1.300000                                                           experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator trained train result train_acc:   0.556000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m1.300000\u001b[0m                                                           \u001b]8;id=492334;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=710415;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.556000\u001b[0m                                                           \u001b]8;id=659849;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=519060;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "validate: 100%|██████████| 27/27 [00:00<00:00, 30.75it/s]\n",
-      "validate:   0%|          | 0/27 [00:00           METRIC   Round 0, collaborator trained train result train_loss:  1.293000                                                           experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m1.293000\u001b[0m                                                           \u001b]8;id=112668;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=805845;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:21] METRIC   Round 0, collaborator validate_local locally_tuned_model_validate result val_acc:       0.741000                           experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator validate_local locally_tuned_model_validate result val_acc:       0.763000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:21]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.741000\u001b[0m                           \u001b]8;id=826398;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=204705;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.763000\u001b[0m                           \u001b]8;id=500589;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=443200;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1620,31 +1643,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator validate_local locally_tuned_model_validate result val_loss:      0.787000                           experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator validate_local locally_tuned_model_validate result val_loss:      0.745000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.787000\u001b[0m                           \u001b]8;id=623019;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=660695;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.745000\u001b[0m                           \u001b]8;id=25970;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=144002;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "validate: 100%|██████████| 27/27 [00:00<00:00, 29.90it/s]\n"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 0, collaborator validate_local locally_tuned_model_validate result val_acc:       0.746000                           experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.746000\u001b[0m                           \u001b]8;id=70335;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=413043;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:22] METRIC   Round 0, collaborator validate_agg aggregated_model_validate result val_acc:    0.125000                                   experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator validate_local locally_tuned_model_validate result val_loss:      0.750000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:22]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.125000\u001b[0m                                   \u001b]8;id=703863;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=272245;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.750000\u001b[0m                           \u001b]8;id=522021;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=759175;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1653,31 +1682,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator validate_agg aggregated_model_validate result val_loss:   2.091000                                   experiment.py:132\n",
+       "
[15:22:59] METRIC   Round 0, collaborator validate_agg aggregated_model_validate result val_acc:    0.125000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m2.091000\u001b[0m                                   \u001b]8;id=620069;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=920869;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:22:59]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.125000\u001b[0m                                   \u001b]8;id=842602;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=899573;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "train:   1%|          | 1/94 [00:00<00:21,  4.23it/s]"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 0, collaborator validate_agg aggregated_model_validate result val_loss:   2.093000                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m2.093000\u001b[0m                                   \u001b]8;id=730453;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=662903;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:24] METRIC   Round 0, collaborator Aggregator train result train_acc:        0.857000                                                   experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator validate_agg aggregated_model_validate result val_acc:    0.150000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:24]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator train result train_acc:        \u001b[1;36m0.857000\u001b[0m                                                   \u001b]8;id=578725;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=359228;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.150000\u001b[0m                                   \u001b]8;id=929637;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=931986;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1686,11 +1721,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator Aggregator train result train_loss:       1.300000                                                   experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator validate_agg aggregated_model_validate result val_loss:   2.089000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator train result train_loss:       \u001b[1;36m1.300000\u001b[0m                                                   \u001b]8;id=821023;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=425687;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m2.089000\u001b[0m                                   \u001b]8;id=407894;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=824562;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1699,11 +1734,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator Aggregator locally_tuned_model_validate result val_acc:   0.741000                                   experiment.py:132\n",
+       "
[15:23:00] METRIC   Round 0, collaborator Aggregator train result train_acc:        0.617505                                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_acc:   \u001b[1;36m0.741000\u001b[0m                                   \u001b]8;id=725270;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=981196;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:00]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator train result train_acc:        \u001b[1;36m0.617505\u001b[0m                                                   \u001b]8;id=277606;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=275814;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1712,11 +1747,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator Aggregator locally_tuned_model_validate result val_loss:  0.787000                                   experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator Aggregator train result train_loss:       1.278999                                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_loss:  \u001b[1;36m0.787000\u001b[0m                                   \u001b]8;id=286807;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=725907;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator train result train_loss:       \u001b[1;36m1.278999\u001b[0m                                                   \u001b]8;id=157511;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=164661;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1725,11 +1760,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator Aggregator aggregated_model_validate result val_acc:      0.125000                                   experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator Aggregator locally_tuned_model_validate result val_acc:   0.754502                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator aggregated_model_validate result val_acc:      \u001b[1;36m0.125000\u001b[0m                                   \u001b]8;id=775225;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=135653;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_acc:   \u001b[1;36m0.754502\u001b[0m                                   \u001b]8;id=606626;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=411576;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1738,32 +1773,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 0, collaborator Aggregator aggregated_model_validate result val_loss:     2.091000                                   experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator Aggregator locally_tuned_model_validate result val_loss:  0.747499                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator aggregated_model_validate result val_loss:     \u001b[1;36m2.091000\u001b[0m                                   \u001b]8;id=688864;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=13591;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_loss:  \u001b[1;36m0.747499\u001b[0m                                   \u001b]8;id=177345;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=568241;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "train: 100%|██████████| 94/94 [00:06<00:00, 14.37it/s]\n",
-      "validate:  26%|██▌       | 7/27 [00:00<00:00, 29.16it/s]"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 0, collaborator Aggregator aggregated_model_validate result val_acc:      0.137496                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator aggregated_model_validate result val_acc:      \u001b[1;36m0.137496\u001b[0m                                   \u001b]8;id=45775;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=775976;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:31] METRIC   Round 1, collaborator trained train result train_acc:   0.714000                                                           experiment.py:132\n",
+       "
           METRIC   Round 0, collaborator Aggregator aggregated_model_validate result val_loss:     2.091001                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:31]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.714000\u001b[0m                                                           \u001b]8;id=89403;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=621180;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m0\u001b[0m, collaborator Aggregator aggregated_model_validate result val_loss:     \u001b[1;36m2.091001\u001b[0m                                   \u001b]8;id=589227;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=121181;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1772,32 +1812,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator trained train result train_loss:  0.717000                                                           experiment.py:132\n",
+       "
[15:23:02] METRIC   Round 1, collaborator trained train result train_acc:   0.815000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m0.717000\u001b[0m                                                           \u001b]8;id=206021;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=451412;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:02]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.815000\u001b[0m                                                           \u001b]8;id=793130;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=482711;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "validate: 100%|██████████| 27/27 [00:00<00:00, 37.86it/s]\n",
-      "validate:  33%|███▎      | 9/27 [00:00<00:00, 24.02it/s]"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 1, collaborator trained train result train_loss:  0.693000                                                           experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m0.693000\u001b[0m                                                           \u001b]8;id=384803;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=218508;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:32] METRIC   Round 1, collaborator validate_local locally_tuned_model_validate result val_acc:       0.801000                           experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator validate_local locally_tuned_model_validate result val_acc:       0.820000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:32]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.801000\u001b[0m                           \u001b]8;id=355547;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=917969;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.820000\u001b[0m                           \u001b]8;id=120665;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=130133;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1806,31 +1851,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator validate_local locally_tuned_model_validate result val_loss:      0.568000                           experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator validate_local locally_tuned_model_validate result val_loss:      0.532000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.568000\u001b[0m                           \u001b]8;id=236060;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=315769;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.532000\u001b[0m                           \u001b]8;id=149339;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=41473;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "validate: 100%|██████████| 27/27 [00:00<00:00, 29.53it/s]\n"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 1, collaborator validate_agg aggregated_model_validate result val_acc:    0.751000                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.751000\u001b[0m                                   \u001b]8;id=772968;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=370425;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:33] METRIC   Round 1, collaborator validate_agg aggregated_model_validate result val_acc:    0.741000                                   experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator validate_agg aggregated_model_validate result val_loss:   0.750000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:33]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.741000\u001b[0m                                   \u001b]8;id=854745;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=748988;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m0.750000\u001b[0m                                   \u001b]8;id=52637;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=667876;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1839,11 +1890,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator validate_agg aggregated_model_validate result val_loss:   0.787000                                   experiment.py:132\n",
+       "
[15:23:11] METRIC   Round 1, collaborator trained train result train_acc:   0.786000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m0.787000\u001b[0m                                   \u001b]8;id=464926;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=112301;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:11]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.786000\u001b[0m                                                           \u001b]8;id=875783;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=62142;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1852,11 +1903,11 @@
     {
      "data": {
       "text/html": [
-       "
[15:03:34] METRIC   Round 1, collaborator Aggregator train result train_acc:        0.714000                                                   experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator trained train result train_loss:  0.699000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:34]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator train result train_acc:        \u001b[1;36m0.714000\u001b[0m                                                   \u001b]8;id=858533;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=521684;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m0.699000\u001b[0m                                                           \u001b]8;id=957709;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=306149;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1865,31 +1916,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator Aggregator train result train_loss:       0.717000                                                   experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator validate_local locally_tuned_model_validate result val_acc:       0.823000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator train result train_loss:       \u001b[1;36m0.717000\u001b[0m                                                   \u001b]8;id=329324;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=372599;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.823000\u001b[0m                           \u001b]8;id=966210;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=163511;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "train:   3%|▎         | 3/94 [00:00<00:11,  8.17it/s]"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 1, collaborator validate_local locally_tuned_model_validate result val_loss:      0.542000                           experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.542000\u001b[0m                           \u001b]8;id=774345;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=404319;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:35] METRIC   Round 1, collaborator Aggregator locally_tuned_model_validate result val_acc:   0.801000                                   experiment.py:132\n",
+       "
[15:23:12] METRIC   Round 1, collaborator validate_agg aggregated_model_validate result val_acc:    0.760000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:35]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_acc:   \u001b[1;36m0.801000\u001b[0m                                   \u001b]8;id=699613;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=143931;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:12]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.760000\u001b[0m                                   \u001b]8;id=179123;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=630821;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1898,11 +1955,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator Aggregator locally_tuned_model_validate result val_loss:  0.568000                                   experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator validate_agg aggregated_model_validate result val_loss:   0.753000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_loss:  \u001b[1;36m0.568000\u001b[0m                                   \u001b]8;id=227169;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=765262;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m0.753000\u001b[0m                                   \u001b]8;id=11807;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=568829;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1911,11 +1968,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator Aggregator aggregated_model_validate result val_acc:      0.741000                                   experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator Aggregator train result train_acc:        0.800499                                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator aggregated_model_validate result val_acc:      \u001b[1;36m0.741000\u001b[0m                                   \u001b]8;id=205909;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=677517;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator train result train_acc:        \u001b[1;36m0.800499\u001b[0m                                                   \u001b]8;id=548949;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=625199;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1924,32 +1981,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 1, collaborator Aggregator aggregated_model_validate result val_loss:     0.787000                                   experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator Aggregator train result train_loss:       0.696000                                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator aggregated_model_validate result val_loss:     \u001b[1;36m0.787000\u001b[0m                                   \u001b]8;id=828946;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=989191;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator train result train_loss:       \u001b[1;36m0.696000\u001b[0m                                                   \u001b]8;id=699999;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=945747;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "train: 100%|██████████| 94/94 [00:06<00:00, 13.71it/s]\n",
-      "validate:   0%|          | 0/27 [00:00           METRIC   Round 1, collaborator Aggregator locally_tuned_model_validate result val_acc:   0.821500                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_acc:   \u001b[1;36m0.821500\u001b[0m                                   \u001b]8;id=753414;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=605838;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:42] METRIC   Round 2, collaborator trained train result train_acc:   0.821000                                                           experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator Aggregator locally_tuned_model_validate result val_loss:  0.537001                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:42]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.821000\u001b[0m                                                           \u001b]8;id=645434;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=906642;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_loss:  \u001b[1;36m0.537001\u001b[0m                                   \u001b]8;id=230814;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=35318;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1958,32 +2020,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator trained train result train_loss:  0.552000                                                           experiment.py:132\n",
+       "
           METRIC   Round 1, collaborator Aggregator aggregated_model_validate result val_acc:      0.755501                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m0.552000\u001b[0m                                                           \u001b]8;id=235565;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=963707;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator aggregated_model_validate result val_acc:      \u001b[1;36m0.755501\u001b[0m                                   \u001b]8;id=930154;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=41984;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "validate: 100%|██████████| 27/27 [00:01<00:00, 20.16it/s]\n",
-      "validate:  52%|█████▏    | 14/27 [00:00<00:00, 26.97it/s]"
-     ]
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 1, collaborator Aggregator aggregated_model_validate result val_loss:     0.751500                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m1\u001b[0m, collaborator Aggregator aggregated_model_validate result val_loss:     \u001b[1;36m0.751500\u001b[0m                                   \u001b]8;id=936815;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=772419;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:44] METRIC   Round 2, collaborator validate_local locally_tuned_model_validate result val_acc:       0.822000                           experiment.py:132\n",
+       "
[15:23:14] METRIC   Round 2, collaborator trained train result train_acc:   0.857000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:44]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.822000\u001b[0m                           \u001b]8;id=919388;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=179770;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:14]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.857000\u001b[0m                                                           \u001b]8;id=856438;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=736003;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -1992,31 +2059,37 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator validate_local locally_tuned_model_validate result val_loss:      0.474000                           experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator trained train result train_loss:  0.557000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.474000\u001b[0m                           \u001b]8;id=893372;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=398115;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m0.557000\u001b[0m                                                           \u001b]8;id=968986;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=356017;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
     },
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "validate: 100%|██████████| 27/27 [00:01<00:00, 26.91it/s]\n"
-     ]
+     "data": {
+      "text/html": [
+       "
[15:23:15] METRIC   Round 2, collaborator validate_local locally_tuned_model_validate result val_acc:       0.842000                           experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m[15:23:15]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.842000\u001b[0m                           \u001b]8;id=608552;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=856962;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
      "data": {
       "text/html": [
-       "
[15:03:45] METRIC   Round 2, collaborator validate_agg aggregated_model_validate result val_acc:    0.801000                                   experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator validate_local locally_tuned_model_validate result val_loss:      0.468000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:45]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.801000\u001b[0m                                   \u001b]8;id=433863;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=447290;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.468000\u001b[0m                           \u001b]8;id=628556;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=279826;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2025,11 +2098,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator validate_agg aggregated_model_validate result val_loss:   0.568000                                   experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator validate_agg aggregated_model_validate result val_acc:    0.824000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m0.568000\u001b[0m                                   \u001b]8;id=299726;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=192999;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.824000\u001b[0m                                   \u001b]8;id=231701;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=361525;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2038,11 +2111,11 @@
     {
      "data": {
       "text/html": [
-       "
[15:03:46] METRIC   Round 2, collaborator Aggregator train result train_acc:        0.821000                                                   experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator validate_agg aggregated_model_validate result val_loss:   0.539000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:46]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator train result train_acc:        \u001b[1;36m0.821000\u001b[0m                                                   \u001b]8;id=830647;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=561967;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m0.539000\u001b[0m                                   \u001b]8;id=231442;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=207650;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2051,11 +2124,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator Aggregator train result train_loss:       0.552000                                                   experiment.py:132\n",
+       "
[15:23:24] METRIC   Round 2, collaborator trained train result train_acc:   0.778000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator train result train_loss:       \u001b[1;36m0.552000\u001b[0m                                                   \u001b]8;id=472704;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=784988;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:24]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator trained train result train_acc:   \u001b[1;36m0.778000\u001b[0m                                                           \u001b]8;id=85217;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=563473;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2064,11 +2137,11 @@
     {
      "data": {
       "text/html": [
-       "
[15:03:47] METRIC   Round 2, collaborator Aggregator locally_tuned_model_validate result val_acc:   0.822000                                   experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator trained train result train_loss:  0.559000                                                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m[15:03:47]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_acc:   \u001b[1;36m0.822000\u001b[0m                                   \u001b]8;id=611658;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=744753;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator trained train result train_loss:  \u001b[1;36m0.559000\u001b[0m                                                           \u001b]8;id=301649;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=440300;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2077,11 +2150,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator Aggregator locally_tuned_model_validate result val_loss:  0.474000                                   experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator validate_local locally_tuned_model_validate result val_acc:       0.837000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_loss:  \u001b[1;36m0.474000\u001b[0m                                   \u001b]8;id=425186;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=167934;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_acc:       \u001b[1;36m0.837000\u001b[0m                           \u001b]8;id=148448;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=602743;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2090,11 +2163,11 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator Aggregator aggregated_model_validate result val_acc:      0.801000                                   experiment.py:132\n",
+       "
           METRIC   Round 2, collaborator validate_local locally_tuned_model_validate result val_loss:      0.453000                           experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator aggregated_model_validate result val_acc:      \u001b[1;36m0.801000\u001b[0m                                   \u001b]8;id=430368;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=265109;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_local locally_tuned_model_validate result val_loss:      \u001b[1;36m0.453000\u001b[0m                           \u001b]8;id=39957;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=32939;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2103,11 +2176,102 @@
     {
      "data": {
       "text/html": [
-       "
           METRIC   Round 2, collaborator Aggregator aggregated_model_validate result val_loss:     0.568000                                   experiment.py:132\n",
+       "
[15:23:25] METRIC   Round 2, collaborator validate_agg aggregated_model_validate result val_acc:    0.827000                                   experiment.py:132\n",
        "
\n"
       ],
       "text/plain": [
-       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator aggregated_model_validate result val_loss:     \u001b[1;36m0.568000\u001b[0m                                   \u001b]8;id=105809;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=686326;file:///home/karan/workspace/fl-tutorial/venv/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+       "\u001b[2;36m[15:23:25]\u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_agg aggregated_model_validate result val_acc:    \u001b[1;36m0.827000\u001b[0m                                   \u001b]8;id=784578;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=731909;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator validate_agg aggregated_model_validate result val_loss:   0.524000                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator validate_agg aggregated_model_validate result val_loss:   \u001b[1;36m0.524000\u001b[0m                                   \u001b]8;id=574401;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=619996;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator Aggregator train result train_acc:        0.817503                                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator train result train_acc:        \u001b[1;36m0.817503\u001b[0m                                                   \u001b]8;id=350506;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=487889;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator Aggregator train result train_loss:       0.558000                                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator train result train_loss:       \u001b[1;36m0.558000\u001b[0m                                                   \u001b]8;id=812796;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=238588;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator Aggregator locally_tuned_model_validate result val_acc:   0.839501                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_acc:   \u001b[1;36m0.839501\u001b[0m                                   \u001b]8;id=410232;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=475457;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator Aggregator locally_tuned_model_validate result val_loss:  0.460502                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator locally_tuned_model_validate result val_loss:  \u001b[1;36m0.460502\u001b[0m                                   \u001b]8;id=153438;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=678231;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator Aggregator aggregated_model_validate result val_acc:      0.825500                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator aggregated_model_validate result val_acc:      \u001b[1;36m0.825500\u001b[0m                                   \u001b]8;id=458888;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=218270;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/html": [
+       "
           METRIC   Round 2, collaborator Aggregator aggregated_model_validate result val_loss:     0.531502                                   experiment.py:132\n",
+       "
\n"
+      ],
+      "text/plain": [
+       "\u001b[2;36m          \u001b[0m\u001b[2;36m \u001b[0mMETRIC   Round \u001b[1;36m2\u001b[0m, collaborator Aggregator aggregated_model_validate result val_loss:     \u001b[1;36m0.531502\u001b[0m                                   \u001b]8;id=457378;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py\u001b\\\u001b[2mexperiment.py\u001b[0m\u001b]8;;\u001b\\\u001b[2m:\u001b[0m\u001b]8;id=620827;file:///home/nesubuntu20nvda/miniconda3/envs/hack_openfl/lib/python3.8/site-packages/openfl/interface/interactive_api/experiment.py#132\u001b\\\u001b[2m132\u001b[0m\u001b]8;;\u001b\\\n"
       ]
      },
      "metadata": {},
@@ -2132,12 +2296,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 22,
    "id": "e5d793be-6c20-4a22-bad7-c082c1ee76ca",
    "metadata": {
     "id": "e5d793be-6c20-4a22-bad7-c082c1ee76ca"
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[None, None]"
+      ]
+     },
+     "execution_count": 22,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
     "# To stop all services run\n",
     "!pkill fx\n",
@@ -2151,7 +2326,7 @@
    "provenance": []
   },
   "kernelspec": {
-   "display_name": "Python 3.8.0 ('venv': venv)",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -2165,7 +2340,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.0"
+   "version": "3.8.16"
   },
   "vscode": {
    "interpreter": {
diff --git a/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb b/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb
index 8aa6844..442d0f9 100644
--- a/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb
+++ b/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb
@@ -8,6 +8,7 @@
    "source": [
     "# Federated GaNDLF Tutorial with PathMNIST Dataset\n",
     "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/intel/fl-tutorial/blob/gh-pages/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb)\n",
+    "### Note: If running the notebook on google colab, please make sure to use python version 3.8, OpenFL works with this version only.\n",
     "\n",
     "In this tutorial, we will be using the **Generally Nuanced Deep Learning Framework** (GaNDLF) to perform training and inference on a ResNet architecture, with PathMNIST, a dataset of colon pathology images, using [OpenFL](https://github.com/intel/openfl) in a federated learning fashion. This is a multi-class classification task: there are 9 different types of colon tissue displayed in the pathology images, each represented by its own class.\n",
     "\n",

From b1320883ec623e5f312619b13fe9b182c2c46b1b Mon Sep 17 00:00:00 2001
From: NeethuES-intel 
Date: Mon, 22 May 2023 22:46:02 -0700
Subject: [PATCH 2/2] fix(bug 766): added python version update

---
 .../Tutorial_1_Medmnist_2D.ipynb              | 17 ++++++++++++++++
 .../Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb   | 20 +++++++++++++++++--
 2 files changed, 35 insertions(+), 2 deletions(-)

diff --git a/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb b/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb
index 1ae69f7..8a38b83 100644
--- a/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb
+++ b/tutorial_notebooks/Tutorial_1_Medmnist_2D.ipynb
@@ -22,6 +22,23 @@
     "Let's get these out of the way first."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "415e25ee",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#If python version != 3.8, then uncomment & run the following code to change the python version to 3.8\n",
+    "!python --version\n",
+    "\n",
+    "#!sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1\n",
+    "#!sudo update-alternatives --config python3\n",
+    "#!python3 --version\n",
+    "#!sudo apt install python3-pip\n",
+    "#!python --version"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
diff --git a/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb b/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb
index 442d0f9..dd715a5 100644
--- a/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb
+++ b/tutorial_notebooks/Tutorial_3_GaNDLF_OpenFL_tutorial.ipynb
@@ -26,6 +26,22 @@
     "Let's get these out of the way first."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#If python version != 3.8, then uncomment & run the following code to change the python version to 3.8\n",
+    "!python --version\n",
+    "\n",
+    "#!sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 1\n",
+    "#!sudo update-alternatives --config python3\n",
+    "#!python3 --version\n",
+    "#!sudo apt install python3-pip\n",
+    "#!python --version"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
@@ -2044,7 +2060,7 @@
   },
   "gpuClass": "standard",
   "kernelspec": {
-   "display_name": "Python 3.8.0 ('venv': venv)",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -2058,7 +2074,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.0"
+   "version": "3.8.16"
   },
   "vscode": {
    "interpreter": {