Added GPU for aggregator

Signed-off-by: Parth Mandaliya <[email protected]>

openfl-tutorials/experimental/Workflow_Interface_201_Exclusive_GPUs_with_Ray.ipynb

@@ -291,7 +291,7 @@
    "source": [
     "In this step we define entities necessary to run the flow and create a function which returns dataset as private attributes of collaborator. As described in [quickstart](https://github.com/securefederatedai/openfl/blob/develop/openfl-tutorials/experimental/Workflow_Interface_101_MNIST.ipynb) we define entities necessary for the flow.\n",
     "\n",
-    "To request GPU(s) with ray-backend, we specify `num_gpus=0.5` as the argument while instantiating Collaborator, this will reserve 0.5 GPU for each of the 2 collaborators and therefore require a dedicated GPU for the exeriment. Tune this based on your use case, for example `num_gpus=0.5` for an experiment with 4 collaborators will require 2 dedicated GPUs. **NOTE:** Collaborator cannot span over multiple GPUs, for example `num_gpus=0.4` with 5 collaborators will require 3 dedicated GPUs. In this case collaborator 1 and 2 use GPU#1, collaborator 3 and 4 use GPU#2, and collaborator 5 uses GPU#3."
+    "To request GPU(s) with ray-backend, we specify `num_gpus=0.3` as the argument while instantiating Aggregator and Collaborator, this will reserve 0.3 GPU for each of the 2 collaborators and the aggregator and therefore require a dedicated GPU for the experiment. Tune this based on your use case, for example `num_gpus=0.4` for an experiment with 4 collaborators and the aggregator will require 2 dedicated GPUs. **NOTE:** Collaborator cannot span over multiple GPUs, for example `num_gpus=0.4` with 5 collaborators will require 3 dedicated GPUs. In this case collaborator 1 and 2 use GPU#1, collaborator 3 and 4 use GPU#2, and collaborator 5 uses GPU#3."
    ]
   },
   {
@@ -302,7 +302,7 @@
    "outputs": [],
    "source": [
     "# Setup Aggregator private attributes via callable function\n",
-    "aggregator = Aggregator()\n",
+    "aggregator = Aggregator(num_gpus=0.3)\n",
     "\n",
     "collaborator_names = ['Portland', 'Seattle']\n",
     "\n",
@@ -325,7 +325,7 @@
     "for idx, collaborator_name in enumerate(collaborator_names):\n",
     "    collaborators.append(\n",
     "        Collaborator(\n",
-    "            name=collaborator_name, num_cpus=0, num_gpus=0.5,\n",
+    "            name=collaborator_name, num_cpus=0, num_gpus=0.3,\n",
     "            private_attributes_callable=callable_to_initialize_collaborator_private_attributes,\n",
     "            index=idx, n_collaborators=len(collaborator_names),\n",
     "            train_dataset=mnist_train, test_dataset=mnist_test, batch_size_train=batch_size_train\n",