diff --git a/beginner_source/blitz/cifar10_tutorial.py b/beginner_source/blitz/cifar10_tutorial.py index 766666eade9..5a9cde3f105 100644 --- a/beginner_source/blitz/cifar10_tutorial.py +++ b/beginner_source/blitz/cifar10_tutorial.py @@ -105,7 +105,7 @@ def imshow(img): # get some random training images dataiter = iter(trainloader) -images, labels = dataiter.next() +images, labels = next(dataiter) # show images imshow(torchvision.utils.make_grid(images)) @@ -210,7 +210,7 @@ def forward(self, x): # Okay, first step. Let us display an image from the test set to get familiar. dataiter = iter(testloader) -images, labels = dataiter.next() +images, labels = next(dataiter) # print images imshow(torchvision.utils.make_grid(images))