src/route_layer.c
@@ -21,11 +21,11 @@ fprintf(stderr, "\n"); l.outputs = outputs; l.inputs = outputs; l.delta = calloc(outputs*batch, sizeof(float)); 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; }