Browse Source

change 1 - keep_prob to keep_prob

tags/v0.5.0-beta
chenzomi 5 years ago
parent
commit
a1eceacc99
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu

+ 1
- 1
mindspore/ccsrc/kernel/gpu/cuda_impl/dropout_impl.cu View File

@@ -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];


Loading…
Cancel
Save