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

Update train.c

Browse Source
This commit is contained in:
augustin64 2022-12-19 15:49:03 +01:00
parent 80e82c16f6
commit c913bf3195
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/cnn/train.c
View File

@ -133,7 +133,7 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
// Création des paramètres donnés à chaque thread dans le cas du multi-threading // Création des paramètres donnés à chaque thread dans le cas du multi-threading
TrainParameters** train_parameters = (TrainParameters**)malloc(sizeof(TrainParameters*)*nb_threads); TrainParameters** train_parameters = (TrainParameters**)malloc(sizeof(TrainParameters*)*nb_threads);
TrainParameters* param; TrainParameters* param;
for (int k=0; k < nb_threads; k++) { for (int k=0; k < nb_threads; k++) {
train_parameters[k] = (TrainParameters*)malloc(sizeof(TrainParameters)); train_parameters[k] = (TrainParameters*)malloc(sizeof(TrainParameters));
param = train_parameters[k]; param = train_parameters[k];
@ -219,7 +219,10 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
// On attend la terminaison de chaque thread un à un // On attend la terminaison de chaque thread un à un
pthread_join( tid[k], NULL ); pthread_join( tid[k], NULL );
accuracy += train_parameters[k]->accuracy / (float) nb_images_total; accuracy += train_parameters[k]->accuracy / (float) nb_images_total;
}
// On attend que tous les fils aient fini avant d'appliquer des modifications au réseau principal
for (int k=0; k < nb_threads; k++) {
update_weights(network, train_parameters[k]->network, train_parameters[k]->nb_images); update_weights(network, train_parameters[k]->network, train_parameters[k]->nb_images);
update_bias(network, train_parameters[k]->network, train_parameters[k]->nb_images); update_bias(network, train_parameters[k]->network, train_parameters[k]->nb_images);
free_network(train_parameters[k]->network); free_network(train_parameters[k]->network);