Created using Colaboratory

Author: istranic

colabs/Getting_Started_with_Hub.ipynb

@@ -240,7 +240,7 @@
       "source": [
         "# Download dataset\n",
         "from IPython.display import clear_output\n",
-        "!wget https://firebasestorage.googleapis.com/v0/b/gitbook-28427.appspot.com/o/assets%2F-M_MXHpa1Cq7qojD2u_r%2F-MbI7YlHiBJg6Fg-HsOf%2F-MbIUlXZn7EYdgDNncOI%2Fanimals.zip?alt=media&token=c491c2cb-7f8b-4b23-9617-a843d38ac611\n",
+        "!wget https://github.com/activeloopai/examples/raw/main/colabs/starting_data/animals.zip\n",
         "clear_output()"
       ],
       "execution_count": null,
@@ -253,7 +253,7 @@
       },
       "source": [
         "# Unzip to './animals' folder\n",
-        "!unzip -qq /content/assets%2F-M_MXHpa1Cq7qojD2u_r%2F-MbI7YlHiBJg6Fg-HsOf%2F-MbIUlXZn7EYdgDNncOI%2Fanimals.zip?alt=media"
+        "!unzip -qq /content/animals.zip"
       ],
       "execution_count": null,
       "outputs": []
@@ -1100,6 +1100,26 @@
       "execution_count": null,
       "outputs": []
     },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "**Note:** Some datasets such as imagenet contain both grayscale and color images, which can cause errors when the transformed images are passed to the model. To convert only the grayscale images to color format, you can add this Torchvision transform to your pipeline:"
+      ],
+      "metadata": {
+        "id": "-EQ2LUPydfPo"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# transforms.Lambda(lambda x: x.repeat(int(3/x.shape[0]), 1, 1))"
+      ],
+      "metadata": {
+        "id": "3Wi0K1aVdjMr"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "markdown",
       "metadata": {