From a1eceacc997be694fbee7b102f364a52f847d9c5 Mon Sep 17 00:00:00 2001
From: chenzomi <chenzomi12@gmail.com>
Date: Mon, 25 May 2020 11:16:35 +0800
Subject: [PATCH] change 1 - keep_prob to keep_prob

---
 mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu b/mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu
index bffa73fb76..940c64ea53 100644
--- 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];