diff options
Diffstat (limited to 'include/weights.h')
| -rw-r--r-- | include/weights.h | 21 |
1 files changed, 6 insertions, 15 deletions
diff --git a/include/weights.h b/include/weights.h index ac5ede8..c4e0c3c 100644 --- a/include/weights.h +++ b/include/weights.h @@ -15,20 +15,11 @@ along with ct. If not, see <https://www.gnu.org/licenses/>. */ -#define KERN_SIZE 3 -#define KERN_CHAN 6 -#define KERN_NUM 12 -#define DENSE1_NUM 11 -#define DENSE2_NUM 8 +#define INP_NUM 3 + 5*5 +#define DENSE_NUM 64 -#define KERN_OSIZE (5-KERN_SIZE+1) // 5 -#define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108 -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[DENSE2_NUM]; -extern const float output_bias; +extern const float dense1_weights[DENSE_NUM][INP_NUM]; +extern const float dense1_biases[DENSE_NUM]; +extern const float output_weights[2][DENSE_NUM]; +extern const float output_bias[2]; |