Fix file loading order of N-MNIST samples

neuromorphs · Jul 9, 2024 · df554e7 · df554e7
1 parent d67aa25
commit df554e7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tonic/datasets/nmnist.py b/tonic/datasets/nmnist.py
@@ -97,6 +97,7 @@ def __init__(
 
         file_path = os.path.join(self.location_on_system, self.folder_name)
         for path, dirs, files in os.walk(file_path):
+            dirs.sort()
             files.sort()
             for file in files:
                 if file.endswith("bin"):