mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Added BFLOPs output for network configurations

Browse Source
This commit is contained in:
AlexeyAB
2018-05-14 13:34:40 +03:00
parent 3ebcc647b6
commit 8b5344ee2d
4 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/convolutional_layer.c
View File

@ -397,7 +397,9 @@ convolutional_layer make_convolutional_layer(int batch, int h, int w, int c, int
l.workspace_size = get_workspace_size(l); l.workspace_size = get_workspace_size(l);
l.activation = activation; l.activation = activation;
fprintf(stderr, "conv %5d %2d x%2d /%2d %4d x%4d x%4d -> %4d x%4d x%4d\n", n, size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c); //fprintf(stderr, "conv %5d %2d x%2d /%2d %4d x%4d x%4d -> %4d x%4d x%4d\n", n, size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c);
l.bflops = (2.0 * l.n * l.size*l.size*l.c * l.out_h*l.out_w) / 1000000000.;
fprintf(stderr, "conv %5d %2d x%2d /%2d %4d x%4d x%4d -> %4d x%4d x%4d %5.3f BF\n", n, size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c, l.bflops);
return l; return l;
} }

1
src/layer.h
View File

@ -113,6 +113,7 @@ struct layer{
int tanh; int tanh;
int *mask; int *mask;
int total; int total;
float bflops;
int adam; int adam;
float B1; float B1;

3
src/maxpool_layer.c
View File

@ -47,7 +47,8 @@ maxpool_layer make_maxpool_layer(int batch, int h, int w, int c, int size, int s
l.output_gpu = cuda_make_array(l.output, output_size); l.output_gpu = cuda_make_array(l.output, output_size);
l.delta_gpu = cuda_make_array(l.delta, output_size); l.delta_gpu = cuda_make_array(l.delta, output_size);
#endif #endif
fprintf(stderr, "max %d x %d / %d %4d x%4d x%4d -> %4d x%4d x%4d\n", size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c); l.bflops = (l.size*l.size*l.c * l.out_h*l.out_w) / 1000000000.;
fprintf(stderr, "max %d x %d / %d %4d x%4d x%4d -> %4d x%4d x%4d %5.3f BF\n", size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c, l.bflops);
return l; return l;
} }

5
src/parser.c
View File

@ -712,6 +712,7 @@ network parse_network_cfg_custom(char *filename, int batch)
params.time_steps = net.time_steps; params.time_steps = net.time_steps;
params.net = net; params.net = net;
float bflops = 0;
size_t workspace_size = 0; size_t workspace_size = 0;
n = n->next; n = n->next;
int count = 0; int count = 0;
@ -719,7 +720,7 @@ network parse_network_cfg_custom(char *filename, int batch)
fprintf(stderr, "layer filters size input output\n"); fprintf(stderr, "layer filters size input output\n");
while(n){ while(n){
params.index = count; params.index = count;
fprintf(stderr, "%5d ", count); fprintf(stderr, "%4d ", count);
s = (section *)n->val; s = (section *)n->val;
options = s->options; options = s->options;
layer l = {0}; layer l = {0};
@ -796,10 +797,12 @@ network parse_network_cfg_custom(char *filename, int batch)
params.c = l.out_c; params.c = l.out_c;
params.inputs = l.outputs; params.inputs = l.outputs;
} }
if (l.bflops > 0) bflops += l.bflops;
} }
free_list(sections); free_list(sections);
net.outputs = get_network_output_size(net); net.outputs = get_network_output_size(net);
net.output = get_network_output(net); net.output = get_network_output(net);
printf("Total BFLOPS %5.3f \n", bflops);
if(workspace_size){ if(workspace_size){
//printf("%ld\n", workspace_size); //printf("%ld\n", workspace_size);
#ifdef GPU #ifdef GPU