src/dropout_layer_kernels.cu
@@ -1,8 +1,11 @@ #include "cuda_runtime.h" #include "curand.h" #include "cublas_v2.h" extern "C" { #include "dropout_layer.h" #include "cuda.h" #include "utils.h" #include "params.h" } __global__ void yoloswag420blazeit360noscope(float *input, int size, float *rand, float prob, float scale)