src/dropout_layer_kernels.cu
@@ -16,6 +16,11 @@ if (!state.train) return; int size = layer.inputs*layer.batch; cuda_random(layer.rand_gpu, size); int i; for(i = 0; i < size; ++i){ layer.rand[i] = rand_uniform(); } cuda_push_array(layer.rand_gpu, layer.rand, size); yoloswag420blazeit360noscope<<<cuda_gridsize(size), BLOCK>>>(state.input, size, layer.rand_gpu, layer.probability, layer.scale); check_error(cudaPeekAtLastError());