intro to pytorch: Part 2, next() is not a method

udacity · Feb 28, 2023 · ff29fe7 · ff29fe7
1 parent c9404fc
commit ff29fe7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Exercises).ipynb
@@ -452,7 +452,7 @@
    "source": [
     "# Grab some data \n",
     "dataiter = iter(trainloader)\n",
-    "images, labels = dataiter.next()\n",
+    "images, labels = next(dataiter)\n",
     "\n",
     "# Resize images into a 1D vector, new shape is (batch size, color channels, image pixels) \n",
     "images.resize_(64, 1, 784)\n",

diff --git a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb
@@ -112,7 +112,7 @@
    ],
    "source": [
     "dataiter = iter(trainloader)\n",
-    "images, labels = dataiter.next()\n",
+    "images, labels = next(dataiter)\n",
     "print(type(images))\n",
     "print(images.shape)\n",
     "print(labels.shape)"