From e858458e934a9e2fca953be43120497771292304 Mon Sep 17 00:00:00 2001 From: 3gg <3gg@shellblade.net> Date: Thu, 12 May 2022 20:31:56 -0700 Subject: Fix typo in OneHotDecode. --- src/bin/mnist/src/main.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src') 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* static int OneHotDecode(const nnMatrix* label_matrix) { assert(label_matrix); - assert(label_matrix->cols == 1); - assert(label_matrix->rows == 10); + assert(label_matrix->cols == 10); + assert(label_matrix->rows == 1); R max_value = 0; int pos_max = 0; -- cgit v1.2.3