提交 948d07ed 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1412 change 1 - keep_prob to keep_prob

Merge pull request !1412 from SanjayChan/master
......@@ -20,7 +20,7 @@
__global__ void DropoutForwardKernel(const float *input, float *mask, float *output, size_t num_count,
float drop_prob) {
float scale = 1.f / (1.f - drop_prob);
float scale = 1.f / drop_prob;
for (size_t i = blockIdx.x * blockDim.x + threadIdx.x; i < num_count; i += blockDim.x * gridDim.x) {
mask[i] = mask[i] > drop_prob;
output[i] = scale * input[i] * mask[i];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册