iter.next() to next(iter)

The new pytorch version does not allow the old command and causes the code to return errors when running. It is a small fix, that will especially help beginners.

Author: danblae

09_dataloader.py

@@ -66,7 +66,7 @@ def __len__(self):
 
 # convert to an iterator and look at one random sample
 dataiter = iter(train_loader)
-data = dataiter.next()
+data = next(dataiter)
 features, labels = data
 print(features, labels)
 
@@ -97,6 +97,6 @@ def __len__(self):
 
 # look at one random sample
 dataiter = iter(train_loader)
-data = dataiter.next()
+data = next(dataiter)
 inputs, targets = data
 print(inputs.shape, targets.shape)

13_feedforward.py

@@ -35,7 +35,7 @@
                                           shuffle=False)
 
 examples = iter(test_loader)
-example_data, example_targets = examples.next()
+example_data, example_targets = next(examples)
 
 for i in range(6):
     plt.subplot(2,3,i+1)

14_cnn.py

@@ -45,7 +45,7 @@ def imshow(img):
 
 # get some random training images
 dataiter = iter(train_loader)
-images, labels = dataiter.next()
+images, labels = next(dataiter)
 
 # show images
 imshow(torchvision.utils.make_grid(images))

16_tensorboard.py

@@ -43,7 +43,7 @@
                                           shuffle=False)
 
 examples = iter(test_loader)
-example_data, example_targets = examples.next()
+example_data, example_targets = next(examples)
 
 for i in range(6):
     plt.subplot(2,3,i+1)