fixed dropout ><

Joseph Redmon

2014-12-13 90d354a2a5a3ba76071337d8794cfc00f7bc5fab

1
2
3
4
5
__kernel void yoloswag420blazeit360noscope(__global float *input, __global float *rand, float prob, float scale)
{
    int id = get_global_id(0);
    input[id] = (rand[id] < prob) ? 0 : input[id]*scale;
}