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

Can validate on imagenet now

Browse Source
This commit is contained in:
Joseph Redmon
2014-11-05 14:49:58 -08:00
parent 2b2441313b
commit b13ad6d5fd
15 changed files with 451 additions and 291 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/softmax_layer.c
View File

@ -50,6 +50,12 @@ void backward_softmax_layer(const softmax_layer layer, float *delta)
}
#ifdef GPU
void pull_softmax_layer_output(const softmax_layer layer)
{
cl_read_array(layer.output_cl, layer.output, layer.inputs*layer.batch);
}
cl_kernel get_softmax_forward_kernel()
{
static int init = 0;
@ -77,6 +83,12 @@ void forward_softmax_layer_gpu(const softmax_layer layer, cl_mem input)
clEnqueueNDRangeKernel(queue, kernel, 1, 0, global_size, 0, 0, 0, 0);
check_error(cl);
/*
cl_read_array(layer.output_cl, layer.output, layer.inputs*layer.batch);
int z;
for(z = 0; z < layer.inputs*layer.batch; ++z) printf("%f,",layer.output[z]);
*/
}
void backward_softmax_layer_gpu(const softmax_layer layer, cl_mem delta)