This commit is contained in:
augustin64 2023-02-28 13:15:16 +01:00
parent 79312caace
commit 1db6c6824d

View File

@ -120,9 +120,9 @@ void free_network(Network* network) {
if (network->kernel[i]->cnn != NULL) { // Convolution if (network->kernel[i]->cnn != NULL) { // Convolution
free_convolution(network, i); free_convolution(network, i);
} else if (network->kernel[i]->nn != NULL) { } else if (network->kernel[i]->nn != NULL) {
if (network->kernel[i]->linearisation == 0) { // Dense non linearised if (network->kernel[i]->linearisation == 0) { // Dense non linearized
free_dense(network, i); free_dense(network, i);
} else { // Dense lineariation } else { // Dense linearisation
free_dense_linearisation(network, i); free_dense_linearisation(network, i);
} }
} else { // Pooling } else { // Pooling