src/dropout_layer_kernels.cu
@@ -14,10 +14,8 @@ extern "C" void forward_dropout_layer_gpu(dropout_layer layer, network_state state) { if (!state.train) return; int j; int size = layer.inputs*layer.batch; for(j = 0; j < size; ++j) layer.rand[j] = rand_uniform(); cuda_push_array(layer.rand_gpu, layer.rand, layer.inputs*layer.batch); cuda_random(layer.rand_gpu, size); yoloswag420blazeit360noscope<<<cuda_gridsize(size), BLOCK>>>(state.input, size, layer.rand_gpu, layer.probability, layer.scale); check_error(cudaPeekAtLastError());