improve documentation of DatasetFolder and ImageFolder (#2112)

pytorch · Apr 16, 2020 · 7b60f4d · 7b60f4d
1 parent 1b9f251
commit 7b60f4d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchvision/datasets/folder.py b/torchvision/datasets/folder.py
@@ -81,7 +81,7 @@ class DatasetFolder(VisionDataset):
             both extensions and is_valid_file should not be passed.
 
      Attributes:
-        classes (list): List of the class names.
+        classes (list): List of the class names sorted alphabetically.
         class_to_idx (dict): Dict with items (class_name, class_index).
         samples (list): List of (sample path, class_index) tuples
         targets (list): The class_index value for each image in the dataset
@@ -193,7 +193,7 @@ class ImageFolder(DatasetFolder):
             and check if the file is a valid file (used to check of corrupt files)
 
      Attributes:
-        classes (list): List of the class names.
+        classes (list): List of the class names sorted alphabetically.
         class_to_idx (dict): Dict with items (class_name, class_index).
         imgs (list): List of (image path, class_index) tuples
     """