augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-24 07:36:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix dataset memory leak

Browse Source
This commit is contained in:
augustin64 2023-01-14 15:28:02 +01:00
parent df48b92cf2
commit 771d69af20
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/cnn/train.c
View File

@ -288,11 +288,26 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
}
free(shuffle_index);
free_network(network);
#ifdef USE_MULTITHREADING
free(tid);
#else
free(train_params);
#endif
if (dataset_type == 0) {
for (int i=0; i < nb_images_total; i++) {
for (int j=0; j < 28; j++) {
free(images[i][j]);
}
free(images[i]);
}
free(images);
free(labels);
} else {
free_dataset(dataset);
}
end_time = omp_get_wtime();
elapsed_time = end_time - algo_start;
printf("\nTemps total: %0.1f s\n", elapsed_time);