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

VadimPushtaev · VadimPushtaev · commit 59f4d56edbed · 2023-02-28T23:27:11.000+02:00
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)"
