From b598a3ab9e682a826516b242a710b26e0c22389f Mon Sep 17 00:00:00 2001 From: tslil Date: Sun, 10 Jan 2021 21:40:38 -0500 Subject: This is the general idea, but i'm sure order is incorrect somewhere --- include/weights.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'include/weights.h') diff --git a/include/weights.h b/include/weights.h index 63ae361..9a6257a 100644 --- a/include/weights.h +++ b/include/weights.h @@ -1,8 +1,8 @@ #define KERN_NUM 12 #define KERN_CHAN 8 #define KERN_SIZE 3 -#define KERN_OSZE (5-KERN_SIZE+1) -#define CONV_NUM (KERN_NUM * KERN_OSZE * KERN_OSZE + 2) +#define KERN_OSIZE (5-KERN_SIZE+1) +#define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE) #define DENSE1_NUM 9 #define DENSE2_NUM 8 @@ -409,7 +409,7 @@ const float conv2d_biases[KERN_NUM] = -0.49402446, -0.44366917, -0.15726285, -0.37746304, 0.03534909, -0.80909145, 0.22329862, -0.16383767}; -const float dense1_weights[CONV_NUM][DENSE1_NUM] = +const float dense1_weights[CONV_NUM+2][DENSE1_NUM] = {{-2.51253784e-01, -1.75899580e-01, 6.41923770e-02, 2.38801807e-01, 1.66776031e-01, -3.04984421e-01, -2.16516569e-01, 1.90220159e-02, 6.49385035e-01}, -- cgit v1.3.1