Change range(len) to enumerate (#2153)

pytorch · Apr 29, 2020 · 1affa2e · 1affa2e
1 parent 9626881
commit 1affa2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/datasets/folder.py b/torchvision/datasets/folder.py
@@ -122,7 +122,7 @@ def _find_classes(self, dir):
         """
         classes = [d.name for d in os.scandir(dir) if d.is_dir()]
         classes.sort()
-        class_to_idx = {classes[i]: i for i in range(len(classes))}
+        class_to_idx = {cls_name: i for i, cls_name in enumerate(classes)}
         return classes, class_to_idx
 
     def __getitem__(self, index):