diff --git a/src/cnn/include/creation.h b/src/cnn/include/creation.h index 1338736..e55f6d0 100644 --- a/src/cnn/include/creation.h +++ b/src/cnn/include/creation.h @@ -18,6 +18,7 @@ Network* create_network_lenet5(float learning_rate, int dropout, int activation, * Renvoie un réseau suivante l'architecture AlexNet * 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