Skip to content

Commit

Permalink
Merge pull request #288 from verilog-indeed/nmnist-fix
Browse files Browse the repository at this point in the history
Fix file loading order of N-MNIST samples
  • Loading branch information
biphasic authored Jul 10, 2024
2 parents d67aa25 + df554e7 commit 5468fbc
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tonic/datasets/nmnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"):
Expand Down

0 comments on commit 5468fbc

Please sign in to comment.