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

Fix some mistakes

Browse Source
This commit is contained in:
julienChemillier 2023-05-14 18:50:52 +02:00
parent 8de03863fa
commit 19757b1c0d
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cnn/creation.c
View File

@ -58,7 +58,7 @@ Network* create_network_alexnet(float learning_rate, int dropout, int activation
add_average_pooling(network, 3, 2, 0); add_average_pooling(network, 3, 2, 0);
add_dense_linearisation(network, 4096, activation); add_dense_linearisation(network, 4096, activation);
add_dense(network, 4096, activation); add_dense(network, 4096, activation);
add_dense(network, size_output, activation); add_dense(network, size_output, SOFTMAX);
return network; return network;
} }
@ -144,6 +144,8 @@ void add_max_pooling(Network* network, int kernel_size, int stride, int padding)
network->kernel[k_pos]->cnn = NULL; network->kernel[k_pos]->cnn = NULL;
network->kernel[k_pos]->nn = NULL; network->kernel[k_pos]->nn = NULL;
network->kernel[k_pos]->stride = stride;
network->kernel[k_pos]->padding = padding;
network->kernel[k_pos]->activation = IDENTITY; // Ne contient pas de fonction d'activation network->kernel[k_pos]->activation = IDENTITY; // Ne contient pas de fonction d'activation
network->kernel[k_pos]->linearisation = DOESNT_LINEARISE; network->kernel[k_pos]->linearisation = DOESNT_LINEARISE;
network->kernel[k_pos]->pooling = MAX_POOLING; network->kernel[k_pos]->pooling = MAX_POOLING;

2
src/cnn/include/creation.h
View File

@ -18,7 +18,7 @@ Network* create_network_lenet5(float learning_rate, int dropout, int activation,
* Renvoie un réseau suivante l'architecture AlexNet * Renvoie un réseau suivante l'architecture AlexNet
* C'est à dire une entrée de 3x227x227 et une sortie de taille 'size_output' * C'est à dire une entrée de 3x227x227 et une sortie de taille 'size_output'
*/ */
Network* create_network_alexnet(float learning_rate, int dropout, int activation, int initialisation, int size_output);
/* /*
* Renvoie un réseau sans convolution, similaire à celui utilisé dans src/dense * Renvoie un réseau sans convolution, similaire à celui utilisé dans src/dense
*/ */