address mnist loading on Windows

hughperkins · Mar 23, 2016 · b32766f · b32766f
1 parent b86b370
commit b32766f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/loaders/MnistLoader.cpp b/src/loaders/MnistLoader.cpp
@@ -52,6 +52,7 @@ STATIC void MnistLoader::load(std::string imagesFilePath, unsigned char *images,
         return;
     }
     string labelsFilePath = replace(imagesFilePath, "-images-idx3-ubyte", "-labels-idx1-ubyte");
+    labelsFilePath = replace(labelsFilePath, "-images.idx3-ubyte", "-labels.idx1-ubyte");
 //    cout << "labelsfilepath: " << labelsFilePath << endl;
 
     fileStartPos = 2 * 4 + (long)startN;