| | |
| | | for (i = 0; i < len; ++i) { |
| | | if (a[i] == ',') ++n; |
| | | } |
| | | for (i = 0; i < n; ++i) { |
| | | for (i = 0; i < n && i < total*2; ++i) { |
| | | float bias = atof(a); |
| | | l.biases[i] = bias; |
| | | a = strchr(a, ',') + 1; |
| | |
| | | for(i = 0; i < len; ++i){ |
| | | if (a[i] == ',') ++n; |
| | | } |
| | | for(i = 0; i < n; ++i){ |
| | | for(i = 0; i < n && i < num*2; ++i){ |
| | | float bias = atof(a); |
| | | l.biases[i] = bias; |
| | | a = strchr(a, ',')+1; |
| | |
| | | params.time_steps = net.time_steps; |
| | | params.net = net; |
| | | |
| | | float bflops = 0; |
| | | size_t workspace_size = 0; |
| | | n = n->next; |
| | | int count = 0; |
| | |
| | | fprintf(stderr, "layer filters size input output\n"); |
| | | while(n){ |
| | | params.index = count; |
| | | fprintf(stderr, "%5d ", count); |
| | | fprintf(stderr, "%4d ", count); |
| | | s = (section *)n->val; |
| | | options = s->options; |
| | | layer l = {0}; |
| | |
| | | params.c = l.out_c; |
| | | params.inputs = l.outputs; |
| | | } |
| | | if (l.bflops > 0) bflops += l.bflops; |
| | | } |
| | | free_list(sections); |
| | | net.outputs = get_network_output_size(net); |
| | | net.output = get_network_output(net); |
| | | printf("Total BFLOPS %5.3f \n", bflops); |
| | | if(workspace_size){ |
| | | //printf("%ld\n", workspace_size); |
| | | #ifdef GPU |