src/route_layer.c
@@ -24,8 +24,8 @@ l.delta = calloc(outputs*batch, sizeof(float)); l.output = calloc(outputs*batch, sizeof(float));; #ifdef GPU l.delta_gpu = cuda_make_array(0, outputs*batch); l.output_gpu = cuda_make_array(0, outputs*batch); l.delta_gpu = cuda_make_array(l.delta, outputs*batch); l.output_gpu = cuda_make_array(l.output, outputs*batch); #endif return l; }