summaryrefslogtreecommitdiff
path: root/include/weights.h
diff options
context:
space:
mode:
authortslil <tslil@posteo.de>2021-01-16 14:08:46 -0500
committertslil <tslil@posteo.de>2026-08-28 19:37:41 +0100
commit50b9955b040005e489127a11cf9c1648438a6266 (patch)
tree1f46f71f6ac9870582f6ec7dd22c8cf4db550753 /include/weights.h
parent0983031d914d5fc391805bdb1034429c2be07383 (diff)
Went back up to 1986 weights
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];