diff --git a/src/dropout_layer.c b/src/dropout_layer.c index 780554fb..25ce98ba 100644 --- a/src/dropout_layer.c +++ b/src/dropout_layer.c @@ -27,7 +27,7 @@ dropout_layer make_dropout_layer(int batch, int inputs, float probability) void resize_dropout_layer(dropout_layer *l, int inputs) { - l->rand = realloc(l->rand, l->inputs*l->batch*sizeof(float)); + l->rand = realloc(l->rand, inputs*l->batch*sizeof(float)); #ifdef GPU cuda_free(l->rand_gpu);