src/dropout_layer.c
@@ -6,7 +6,6 @@ dropout_layer make_dropout_layer(int batch, int inputs, float probability) { fprintf(stderr, "Dropout Layer: %d inputs, %f probability\n", inputs, probability); dropout_layer l = {0}; l.type = DROPOUT; l.probability = probability; @@ -22,6 +21,7 @@ l.backward_gpu = backward_dropout_layer_gpu; l.rand_gpu = cuda_make_array(l.rand, inputs*batch); #endif fprintf(stderr, "dropout p = %.2f %4d -> %4d\n", probability, inputs, inputs); return l; }