Optimize DataLoader iteration in WrappedDataLoader

Author: HemanthSai7

Diff for: beginner_source/nn_tutorial.py

@@ -790,8 +790,7 @@ def __len__(self):
         return len(self.dl)
 
     def __iter__(self):
-        batches = iter(self.dl)
-        for b in batches:
+        for b in self.dl:
             yield (self.func(*b))
 
 train_dl, valid_dl = get_data(train_ds, valid_ds, bs)