src/dropout_layer.c
@@ -1,5 +1,4 @@ #include "dropout_layer.h" #include "params.h" #include "utils.h" #include "cuda.h" #include <stdlib.h> @@ -37,7 +36,7 @@ int i; if (!state.train) return; for(i = 0; i < l.batch * l.inputs; ++i){ float r = rand_uniform(); float r = rand_uniform(0, 1); l.rand[i] = r; if(r < l.probability) state.input[i] = 0; else state.input[i] *= l.scale;