| | |
| | | |
| | | cost_layer make_cost_layer(int batch, int inputs, COST_TYPE cost_type, float scale) |
| | | { |
| | | fprintf(stderr, "Cost Layer: %d inputs\n", inputs); |
| | | fprintf(stderr, "cost %4d\n", inputs); |
| | | cost_layer l = {0}; |
| | | l.type = COST; |
| | | |
| | |
| | | l.delta = calloc(inputs*batch, sizeof(float)); |
| | | l.output = calloc(inputs*batch, sizeof(float)); |
| | | l.cost = calloc(1, sizeof(float)); |
| | | |
| | | l.forward = forward_cost_layer; |
| | | l.backward = backward_cost_layer; |
| | | #ifdef GPU |
| | | l.forward_gpu = forward_cost_layer_gpu; |
| | | l.backward_gpu = backward_cost_layer_gpu; |
| | | |
| | | l.delta_gpu = cuda_make_array(l.output, inputs*batch); |
| | | l.output_gpu = cuda_make_array(l.delta, inputs*batch); |
| | | #endif |