diff options
Diffstat (limited to 'include/weights.h')
| -rw-r--r-- | include/weights.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/include/weights.h b/include/weights.h index 18079ba..b9cc509 100644 --- a/include/weights.h +++ b/include/weights.h @@ -5,7 +5,7 @@ #define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108 #define DENSE1_NUM 9 -#define DENSE2_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]; |