mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
CVPR prep
This commit is contained in:
@ -71,8 +71,6 @@ void binarize_filters_gpu(float *filters, int n, int size, float *binary)
|
||||
|
||||
void forward_convolutional_layer_gpu(convolutional_layer l, network_state state)
|
||||
{
|
||||
int i;
|
||||
|
||||
fill_ongpu(l.outputs*l.batch, 0, l.output_gpu, 1);
|
||||
if(l.binary){
|
||||
binarize_filters_gpu(l.filters_gpu, l.n, l.c*l.size*l.size, l.binary_filters_gpu);
|
||||
@ -103,6 +101,7 @@ void forward_convolutional_layer_gpu(convolutional_layer l, network_state state)
|
||||
l.output_gpu);
|
||||
|
||||
#else
|
||||
int i;
|
||||
int m = l.n;
|
||||
int k = l.size*l.size*l.c;
|
||||
int n = l.out_w*l.out_h;
|
||||
|
Reference in New Issue
Block a user