From 50b9955b040005e489127a11cf9c1648438a6266 Mon Sep 17 00:00:00 2001 From: tslil Date: Sat, 16 Jan 2021 14:08:46 -0500 Subject: Went back up to 1986 weights --- include/weights.h | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'include/weights.h') diff --git a/include/weights.h b/include/weights.h index 71d18da..dd479e3 100644 --- a/include/weights.h +++ b/include/weights.h @@ -1,12 +1,12 @@ -#define KERN_NUM 12 -#define KERN_CHAN 6 #define KERN_SIZE 3 +#define KERN_CHAN 6 +#define KERN_NUM 12 +#define DENSE1_NUM 11 +#define DENSE2_NUM 8 + #define KERN_OSIZE (5-KERN_SIZE+1) // 5 #define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108 -#define DENSE1_NUM 9 -#define DENSE2_NUM 10 - extern const float conv2d_weights[KERN_NUM][KERN_SIZE][KERN_SIZE][KERN_CHAN]; extern const float conv2d_biases[KERN_NUM]; extern const float dense1_weights[DENSE1_NUM][CONV_NUM+2]; -- cgit v1.2.3