Skip to content

Commit b32766f

Browse files
committed
address mnist loading on Windows
1 parent b86b370 commit b32766f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/loaders/MnistLoader.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ STATIC void MnistLoader::load(std::string imagesFilePath, unsigned char *images,
5252
return;
5353
}
5454
string labelsFilePath = replace(imagesFilePath, "-images-idx3-ubyte", "-labels-idx1-ubyte");
55+
labelsFilePath = replace(labelsFilePath, "-images.idx3-ubyte", "-labels.idx1-ubyte");
5556
// cout << "labelsfilepath: " << labelsFilePath << endl;
5657

5758
fileStartPos = 2 * 4 + (long)startN;

0 commit comments

Comments
 (0)