diff --git a/src/cnn/creation.c b/src/cnn/creation.c index 79d27e7..234f889 100644 --- a/src/cnn/creation.c +++ b/src/cnn/creation.c @@ -103,7 +103,7 @@ void add_2d_average_pooling(Network* network, int dim_output) { network->kernel[k_pos]->cnn = NULL; network->kernel[k_pos]->nn = NULL; network->kernel[k_pos]->activation = IDENTITY; // Ne contient pas de fonction d'activation - network->kernel[k_pos]->linearisation = kernel_size; + network->kernel[k_pos]->linearisation = 0; create_a_cube_input_layer(network, n, network->depth[n-1], network->width[n-1]/2); create_a_cube_input_z_layer(network, n, network->depth[n-1], network->width[n-1]/2); // Will it be used ? network->size++; diff --git a/src/cnn/include/struct.h b/src/cnn/include/struct.h index 3dfd529..619b7a7 100644 --- a/src/cnn/include/struct.h +++ b/src/cnn/include/struct.h @@ -24,7 +24,7 @@ typedef struct Kernel { Kernel_cnn* cnn; // NULL si ce n'est pas un cnn Kernel_nn* nn; // NULL si ce n'est pas un nn int activation; // Vaut l'identifiant de la fonction d'activation - int linearisation; // Vaut 1 si c'est la linéarisation d'une couche, 0 sinon ?? Ajouter dans les autres + int linearisation; // Vaut 1 si c'est la linéarisation d'une couche, 0 sinon } Kernel;