Added: calc_anchors

This commit is contained in:
AlexeyAB
2018-03-04 03:09:40 +03:00
parent fe44d3d0f2
commit f39160f6e8
4 changed files with 104 additions and 3 deletions

View File

@ -110,11 +110,9 @@ void backward_reorg_layer_gpu(layer l, network_state state)
{
if (l.reverse) {
reorg_ongpu(l.delta_gpu, l.out_w, l.out_h, l.out_c, l.batch, l.stride, 0, state.delta);
//reorg_ongpu(l.delta_gpu, l.w, l.h, l.c, l.batch, l.stride, 0, state.delta);
}
else {
reorg_ongpu(l.delta_gpu, l.out_w, l.out_h, l.out_c, l.batch, l.stride, 1, state.delta);
//reorg_ongpu(l.delta_gpu, l.w, l.h, l.c, l.batch, l.stride, 1, state.delta);
}
}
#endif