Fix linearisation variable

This commit is contained in:
julienChemillier 2023-01-11 12:09:41 +01:00
parent 1a6eb2d7c7
commit c914370103
2 changed files with 2 additions and 2 deletions

View File

@ -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++;

View File

@ -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;