Update next(iter) to align with Python3 (#2087)

* Update next(iter) to align with Python3
pytorch · Oct 18, 2022 · 7d8cb43 · 7d8cb43
1 parent 998c026
commit 7d8cb43
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/beginner_source/introyt/introyt1_tutorial.py b/beginner_source/introyt/introyt1_tutorial.py
@@ -369,7 +369,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))
@@ -446,7 +446,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))

diff --git a/beginner_source/introyt/tensorboardyt_tutorial.py b/beginner_source/introyt/tensorboardyt_tutorial.py
@@ -115,7 +115,7 @@ def matplotlib_imshow(img, one_channel=False):
 
 # Extract a batch of 4 images
 dataiter = iter(training_loader)
-images, labels = dataiter.next()
+images, labels = next(dataiter)
 
 # Create a grid from the images and show them
 img_grid = torchvision.utils.make_grid(images)
@@ -242,7 +242,7 @@ def forward(self, x):
 
 # Again, grab a single mini-batch of images
 dataiter = iter(training_loader)
-images, labels = dataiter.next()
+images, labels = next(dataiter)
 
 # add_graph() will trace the sample input through your model,
 # and render it as a graph.

diff --git a/beginner_source/introyt/trainingyt.py b/beginner_source/introyt/trainingyt.py
@@ -112,7 +112,7 @@ def matplotlib_imshow(img, one_channel=False):
         plt.imshow(np.transpose(npimg, (1, 2, 0)))
 
 dataiter = iter(training_loader)
-images, labels = dataiter.next()
+images, labels = next(dataiter)
 
 # Create a grid from the images and show them
 img_grid = torchvision.utils.make_grid(images)

diff --git a/intermediate_source/tensorboard_tutorial.rst b/intermediate_source/tensorboard_tutorial.rst
@@ -144,7 +144,7 @@ using `make_grid <https://pytorch.org/vision/stable/utils.html#torchvision.utils
 
     # get some random training images
     dataiter = iter(trainloader)
-    images, labels = dataiter.next()
+    images, labels = next(dataiter)
 
     # create grid of images
     img_grid = torchvision.utils.make_grid(images)