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