mirror of https://github.com/pjreddie/darknet.git
Merge 4ce51a7e70
into f6afaabcdf
This commit is contained in:
commit
8f0ef67df2
|
@ -160,7 +160,6 @@ void forward_region_layer(const layer l, network net)
|
|||
int i,j,b,t,n;
|
||||
memcpy(l.output, net.input, l.outputs*l.batch*sizeof(float));
|
||||
|
||||
#ifndef GPU
|
||||
for (b = 0; b < l.batch; ++b){
|
||||
for(n = 0; n < l.n; ++n){
|
||||
int index = entry_index(l, b, n*l.w*l.h, 0);
|
||||
|
@ -183,7 +182,6 @@ void forward_region_layer(const layer l, network net)
|
|||
int index = entry_index(l, 0, 0, l.coords + !l.background);
|
||||
softmax_cpu(net.input + index, l.classes + l.background, l.batch*l.n, l.inputs/l.n, l.w*l.h, 1, l.w*l.h, 1, l.output + index);
|
||||
}
|
||||
#endif
|
||||
|
||||
memset(l.delta, 0, l.outputs * l.batch * sizeof(float));
|
||||
if(!net.train) return;
|
||||
|
|
Loading…
Reference in New Issue