Fix Task 4 comment

Author: ma595

exercises/01_penguin_classification.ipynb

@@ -267,7 +267,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Apply transforms to the data. See Task 4 exercise comments above.\n",
+    "# Apply transforms we need to PenguinDataset to convert input data and target class to tensors. \n",
+    "# See Task 4 exercise comments above.\n",
     "\n",
     "# Create train_set\n",
     "\n",

worked-solutions/01_penguin_classification_solutions.ipynb

@@ -371,8 +371,9 @@
     }
    ],
    "source": [
-    "# Apply the transforms we need to the PenguinDataset to get out input\n",
-    "# targets as Tensors. See Task 4 exercise comments above.\n",
+    "# Apply transforms we need to PenguinDataset to convert input data and target class to tensors. \n",
+    "# See Task 4 exercise comments above.\n",
+    "\n",
     "\n",
     "# Create train_set\n",
     "train_set = PenguinDataset(\n",