summaryrefslogtreecommitdiff
path: root/include/weights.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/weights.h')
-rw-r--r--include/weights.h10
1 files changed, 5 insertions, 5 deletions
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];