!1412 change 1 - keep_prob to keep_prob

Merge pull request !1412 from SanjayChan/master
5 years ago · 948d07edcd
parent 929ef67bfc a1eceacc99
commit 948d07edcd
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu
+++ b/mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu
@ -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];