From b8545dec7bf22ba591924ae4d905599d6958acbb Mon Sep 17 00:00:00 2001 From: noqqaqq Date: Thu, 1 Jun 2023 14:19:57 +0200 Subject: [PATCH] Enumerate over dataset instead of simple loop --- beginner_source/data_loading_tutorial.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/beginner_source/data_loading_tutorial.py b/beginner_source/data_loading_tutorial.py index 322d9b3009c..0b719ec2a68 100644 --- a/beginner_source/data_loading_tutorial.py +++ b/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: