augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-03-14 06:45:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Rectification of a d_network leak

Browse Source
This commit is contained in:
julienChemillier 2023-05-28 09:12:52 +02:00
parent d7d5a7ae6e
commit d916c6c86b
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/cnn/free.c
View File

@ -208,8 +208,8 @@ void free_d_convolution(Network* network, int pos) {
} }
void free_d_dense(Network* network, int pos) { void free_d_dense(Network* network, int pos) {
D_Kernel_nn* d_k_pos = d_network->kernel[pos]->nn;
D_Network* d_network = network->d_network; D_Network* d_network = network->d_network;
D_Kernel_nn* d_k_pos = d_network->kernel[pos]->nn;
int dim = network->kernel[pos]->nn->size_input; int dim = network->kernel[pos]->nn->size_input;
for (int i=0; i < dim; i++) { for (int i=0; i < dim; i++) {
gree(d_k_pos->d_weights[i], true); gree(d_k_pos->d_weights[i], true);
@ -232,8 +232,8 @@ void free_d_dense(Network* network, int pos) {
} }
void free_d_dense_linearisation(Network* network, int pos) { void free_d_dense_linearisation(Network* network, int pos) {
D_Kernel_nn* d_k_pos = d_network->kernel[pos]->nn;
D_Network* d_network = network->d_network; D_Network* d_network = network->d_network;
D_Kernel_nn* d_k_pos = d_network->kernel[pos]->nn;
int dim = network->kernel[pos]->nn->size_input; int dim = network->kernel[pos]->nn->size_input;
if (network->finetuning <= NN_AND_LINEARISATION) { if (network->finetuning <= NN_AND_LINEARISATION) {
@ -260,7 +260,8 @@ void free_d_dense_linearisation(Network* network, int pos) {
gree(d_k_pos, true); gree(d_k_pos, true);
} }
void free_d_network_creation(Network* network, D_Network* d_network) { void free_d_network(Network* network) {
D_Network* d_network = network->d_network;
for (int i=0; i < network->max_size-1; i++) { for (int i=0; i < network->max_size-1; i++) {
D_Kernel* d_k_i = d_network->kernel[i]; D_Kernel* d_k_i = d_network->kernel[i];
if (d_k_i->cnn) { // Convolution if (d_k_i->cnn) { // Convolution

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

@ -65,6 +65,6 @@ void free_d_dense_linearisation(Network* network, int pos);
/* /*
* Libère entièrement l'espace mémoire alloué dans 'create_d_network' (creation.c) * Libère entièrement l'espace mémoire alloué dans 'create_d_network' (creation.c)
*/ */
void free_d_network_creation(Network* network, D_Network* d_network); void free_d_network(Network* network);
#endif #endif

1
src/cnn/train.c
View File

@ -463,6 +463,7 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
write_network(out, network); write_network(out, network);
} }
free(shuffle_index); free(shuffle_index);
free_d_network(network);
free_network(network); free_network(network);
#ifdef USE_MULTITHREADING #ifdef USE_MULTITHREADING