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/bin')
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