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,