diff options
| author | tslil <tslil@posteo.de> | 2021-01-12 15:51:48 -0500 |
|---|---|---|
| committer | tslil <tslil@posteo.de> | 2026-08-28 19:37:41 +0100 |
| commit | a96ec5a74093eb800d14e143b81302d3e0905b85 (patch) | |
| tree | e555252fe9003b8bb0ae648e031c0fe90d8e72bc /include/weights.h | |
| parent | 67dd4ba3a9adc6a9db8eb543480e9e82310eeccb (diff) | |
one output (of course), Pade approximant of logistic for activation
Diffstat (limited to 'include/weights.h')
| -rw-r--r-- | include/weights.h | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/include/weights.h b/include/weights.h index 3c83010..18079ba 100644 --- a/include/weights.h +++ b/include/weights.h @@ -5,8 +5,7 @@ #define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108 #define DENSE1_NUM 9 -#define DENSE2_NUM 8 -#define OUTPUT_NUM 2 +#define DENSE2_NUM 9 extern const float conv2d_weights[KERN_NUM][KERN_SIZE][KERN_SIZE][KERN_CHAN]; extern const float conv2d_biases[KERN_NUM]; @@ -14,5 +13,5 @@ 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]; +extern const float output_weights[DENSE2_NUM]; +extern const float output_bias; |