Enumerate over dataset instead of simple loop (#2407)

Co-authored-by: noqqaqq <noqqaqq@users.noreply.github.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

Author: noqqaqq

beginner_source/data_loading_tutorial.py

@@ -165,9 +165,7 @@ def __getitem__(self, idx):
 
 fig = plt.figure()
 
-for i in range(len(face_dataset)):
-    sample = face_dataset[i]
-
+for i, sample in enumerate(face_dataset):
     print(i, sample['image'].shape, sample['landmarks'].shape)
 
     ax = plt.subplot(1, 4, i + 1)
@@ -356,9 +354,7 @@ def __call__(self, sample):
                                                ToTensor()
                                            ]))
 
-for i in range(len(transformed_dataset)):
-    sample = transformed_dataset[i]
-
+for i, sample in enumerate(transformed_dataset):
     print(i, sample['image'].size(), sample['landmarks'].size())
 
     if i == 3: