Create `ImageFolder` with the same (sorted) order on different machines (#193)

08b1f59f · Jiren Jin · Soumith Chintala · c1f88ef1 · 08b1f59f
Commit 08b1f59f authored Jul 02, 2017 by Jiren Jin Committed by Soumith Chintala Jul 01, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/folder.py torchvision/datasets/folder.py +1 -1

No files found.
--- a/torchvision/datasets/folder.py
+++ b/torchvision/datasets/folder.py
@@ -24,7 +24,7 @@ def find_classes(dir):
 def make_dataset(dir, class_to_idx):
    images = []
    dir = os.path.expanduser(dir)
-    for target in os.listdir(dir):
+    for target in sorted(os.listdir(dir)):
        d = os.path.join(dir, target)
        if not os.path.isdir(d):
            continue