src/dropout_layer.c
@@ -15,7 +15,11 @@ l.batch = batch; l.rand = calloc(inputs*batch, sizeof(float)); l.scale = 1./(1.-probability); l.forward = forward_dropout_layer; l.backward = backward_dropout_layer; #ifdef GPU l.forward_gpu = forward_dropout_layer_gpu; l.backward_gpu = backward_dropout_layer_gpu; l.rand_gpu = cuda_make_array(l.rand, inputs*batch); #endif return l;