From 2fb71fcbc594c3a3f32ee0c95be2e4e9b9ae97df Mon Sep 17 00:00:00 2001 From: tslil Date: Tue, 12 Jan 2021 00:10:12 -0500 Subject: This finally works! --- include/weights.h | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 include/weights.h (limited to 'include/weights.h') diff --git a/include/weights.h b/include/weights.h new file mode 100644 index 0000000..3c83010 --- /dev/null +++ b/include/weights.h @@ -0,0 +1,18 @@ +#define KERN_NUM 12 +#define KERN_CHAN 8 +#define KERN_SIZE 3 +#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 8 +#define OUTPUT_NUM 2 + +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]; +extern const float dense1_biases[DENSE1_NUM]; +extern const float dense2_weights[DENSE2_NUM][DENSE1_NUM]; +extern const float dense2_biases[DENSE2_NUM]; +extern const float output_weights[OUTPUT_NUM][DENSE2_NUM]; +extern const float output_biases[2]; -- cgit v1.3.1