diff --git a/PyTorch_tutorial.ipynb b/PyTorch_tutorial.ipynb
index ba10a2884acd94b5c4454d443cc12658d4a46aaa..dfb48428de22a835cd4b31b4c9ab75572bfac349 100644
--- a/PyTorch_tutorial.ipynb
+++ b/PyTorch_tutorial.ipynb
@@ -180,6 +180,7 @@
     "\n",
     "num_epochs = 10\n",
     "\n",
+    "# For each epoch, calculate the loss\n",
     "for epoch in range(num_epochs):\n",
     "    total_loss = 0.0\n",
     "    for i, (images, labels) in enumerate(train_loader):\n",
@@ -244,7 +245,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -258,7 +259,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.1"
+   "version": "3.9.12"
   }
  },
  "nbformat": 4,