augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-23 23:26:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue in lenet5

Browse Source
This commit is contained in:
Julien Chemillier 2022-10-03 10:06:31 +02:00
parent a604c96476
commit 659463e4b7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cnn/creation.c
View File

@ -33,7 +33,7 @@ Network* create_network_lenet5(int learning_rate, int dropout, int activation, i
Network* network = create_network(8, learning_rate, dropout, initialisation, input_dim, input_depth);
network->kernel[0]->activation = activation;
network->kernel[0]->linearisation = 0;
add_convolution(network, 1, 32, 6, 28, activation);
add_convolution(network, input_depth, input_dim, 6, 28, activation);
add_2d_average_pooling(network, 28, 14);
add_convolution(network, 6, 14, 16, 10, activation);
add_2d_average_pooling(network, 10, 5);