Update test/cnn_structure.c

This commit is contained in:
augustin64 2023-01-13 15:59:04 +01:00
parent df39c2c299
commit 7aa4ab9403

View File

@ -21,7 +21,6 @@ int main() {
kernel = network->kernel[i]; kernel = network->kernel[i];
if ((!kernel->cnn)&&(!kernel->nn)) { if ((!kernel->cnn)&&(!kernel->nn)) {
printf("\n==== Couche %d de type "YELLOW"Pooling"RESET" ====\n", i); printf("\n==== Couche %d de type "YELLOW"Pooling"RESET" ====\n", i);
printf("Linéarisation: %d\n", kernel->linearisation);
} else if (!kernel->cnn) { } else if (!kernel->cnn) {
printf("\n==== Couche %d de type "GREEN"NN"RESET" ====\n", i); printf("\n==== Couche %d de type "GREEN"NN"RESET" ====\n", i);
printf("input: %d\n", kernel->nn->input_units); printf("input: %d\n", kernel->nn->input_units);
@ -32,8 +31,12 @@ int main() {
printf("rows: %d\n", kernel->cnn->rows); printf("rows: %d\n", kernel->cnn->rows);
printf("columns: %d\n", kernel->cnn->columns); printf("columns: %d\n", kernel->cnn->columns);
} }
if (kernel->linearisation) {
printf(YELLOW"Linéarisation: %d\n"RESET, kernel->linearisation);
}
printf("width: %d\n", network->width[i]); printf("width: %d\n", network->width[i]);
printf("depth: %d\n", network->depth[i]); printf("depth: %d\n", network->depth[i]);
printf("activation: %d\n", kernel->activation);
} }
printf(GREEN "\nOK\n" RESET); printf(GREEN "\nOK\n" RESET);