reduce dataset proportion for ogb training on pod4 (#46)

* reduce dataset proportion for ogb training on pod4

* precommit

Author: rahult-graphcore

ogb-competition/gps++_training.ipynb

@@ -33,7 +33,7 @@
    "source": [
     "#### Running on Paperspace\n",
     "\n",
-    "> Currently, this notebook requires a POD16 machine type to run successfully.\n",
+    "> This notebook is designed to be run on a POD16. If you are using a POD4 we will reduce the proportion of the dataset used to enable it to run. For training on the full dataset we recommend using a POD16.\n",
     "\n",
     "The Paperspace environment lets you run this notebook with no set up. To improve your experience we preload datasets and pre-install packages, this can take a few minutes, if you experience errors immediately after starting a session please try restarting the kernel before contacting support. If a problem persists or you want to give us feedback on the content of this notebook, please reach out to through our community of developers using our [slack channel](https://www.graphcore.ai/join-community) or raise a [GitHub issue](https://github.com/gradient-ai/Graphcore-Tensorflow2/issues)."
    ]
@@ -245,6 +245,24 @@
     "    wandb.init(entity=cfg.wandb_entity, project=cfg.wandb_project, config=cfg.as_dict())"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "This notebook is designed to be run on a POD16. If you are using a POD4 we will reduce the proportion of the dataset used to enable it to run. For training on the full dataset we recommend using a POD16."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "pod_type = os.getenv(\"GRAPHCORE_POD_TYPE\", \"pod16\")\n",
+    "if pod_type == \"pod4\":\n",
+    "    cfg.dataset.prop_to_use = 0.2"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},