From f4c35a745dd436a8c38e187f565597248266d9d8 Mon Sep 17 00:00:00 2001 From: Yuxin Wu Date: Thu, 26 Jan 2023 10:36:34 -0800 Subject: [PATCH] Update cifar10_tutorial.py --- beginner_source/blitz/cifar10_tutorial.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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))