Some fixes to momentum

Joseph Redmon

2014-12-08 a99050f0c8cb0315fa31e3d1fa3e38594fe5e40a

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