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

udacity · Mar 2, 2023 · 78e3873 · 78e3873
1 parent c9404fc
commit 78e3873
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 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)"
@@ -650,7 +650,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 4 - Fashion-MNIST (Solution).ipynb b/intro-to-pytorch/Part 4 - Fashion-MNIST (Solution).ipynb
@@ -207,7 +207,7 @@
     "# Test out your network!\n",
     "\n",
     "dataiter = iter(testloader)\n",
-    "images, labels = dataiter.next()\n",
+    "images, labels = next(dataiter)\n",
     "img = images[1]\n",
     "\n",
     "# TODO: Calculate the class probabilities (softmax) for img\n",

diff --git a/intro-to-pytorch/Part 5 - Inference and Validation (Exercises).ipynb b/intro-to-pytorch/Part 5 - Inference and Validation (Exercises).ipynb
@@ -367,7 +367,7 @@
     "model.eval()\n",
     "\n",
     "dataiter = iter(testloader)\n",
-    "images, labels = dataiter.next()\n",
+    "images, labels = next(dataiter)\n",
     "img = images[0]\n",
     "# Convert 2D image to 1D vector\n",
     "img = img.view(1, 784)\n",

diff --git a/intro-to-pytorch/Part 5 - Inference and Validation (Solution).ipynb b/intro-to-pytorch/Part 5 - Inference and Validation (Solution).ipynb
@@ -507,7 +507,7 @@
     "model.eval()\n",
     "\n",
     "dataiter = iter(testloader)\n",
-    "images, labels = dataiter.next()\n",
+    "images, labels = next(dataiter)\n",
     "img = images[0]\n",
     "# Convert 2D image to 1D vector\n",
     "img = img.view(1, 784)\n",