aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/bin/mnist/src/main.c4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/bin/mnist/src/main.c b/src/bin/mnist/src/main.c
index 4d268ac..9aa3ce5 100644
--- a/src/bin/mnist/src/main.c
+++ b/src/bin/mnist/src/main.c
@@ -252,8 +252,8 @@ static void OneHotEncode(const uint8_t* labels_bytes, int num_labels, nnMatrix*
252 252
253static int OneHotDecode(const nnMatrix* label_matrix) { 253static int OneHotDecode(const nnMatrix* label_matrix) {
254 assert(label_matrix); 254 assert(label_matrix);
255 assert(label_matrix->cols == 1); 255 assert(label_matrix->cols == 10);
256 assert(label_matrix->rows == 10); 256 assert(label_matrix->rows == 1);
257 257
258 R max_value = 0; 258 R max_value = 0;
259 int pos_max = 0; 259 int pos_max = 0;