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

Update train.c

Browse Source
This commit is contained in:
augustin64 2023-01-17 12:50:35 +01:00
parent 6917711271
commit 61468044a9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cnn/train.c
View File

@ -46,7 +46,7 @@ void* train_thread(void* parameters) {
write_image_in_network_32(images[index[i]], height, width, network->input[0][0]); write_image_in_network_32(images[index[i]], height, width, network->input[0][0]);
forward_propagation(network); forward_propagation(network);
maxi = indice_max(network->input[network->size-1][0][0], 10); maxi = indice_max(network->input[network->size-1][0][0], 10);
backward_propagation(network, labels[i]); backward_propagation(network, labels[index[i]]);
if (maxi == labels[index[i]]) { if (maxi == labels[index[i]]) {
accuracy += 1.; accuracy += 1.;
@ -294,7 +294,7 @@ void train(int dataset_type, char* images_file, char* labels_file, char* data_di
#else #else
free(train_params); free(train_params);
#endif #endif
if (dataset_type == 0) { if (dataset_type == 0) {
for (int i=0; i < nb_images_total; i++) { for (int i=0; i < nb_images_total; i++) {
for (int j=0; j < 28; j++) { for (int j=0; j < 28; j++) {