Fix issue #75 - Andrew Violette tracked down the bug with the MNIST l…

…oader not working correctly - was copying byte sized blocks, not float-sized blocks
migueldeicaza · Nov 8, 2017 · 25e35d1 · 25e35d1
1 parent 9b32ca9
commit 25e35d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Learn/Datasets/MNIST.cs b/Learn/Datasets/MNIST.cs
@@ -68,8 +68,8 @@ internal BatchReader (MnistImage [] source, byte [] labels, byte [,] oneHotLabel
 
 				int p = 0;
 				for (int item = 0; item < batchSize; item++) {
-					Buffer.BlockCopy (source [start+item].DataFloat, 0, imageData, p, 784);
-					p += 784;
+					Buffer.BlockCopy (source [start+item].DataFloat, 0, imageData, p, 784 * sizeof (float));
+					p += 784 * sizeof (float);
 					for (var j = 0; j < 10; j++)
 						labelData [item, j] = oneHotLabels [item + start, j];
 				}