Merge pull request #9325 from dzhwinter/fix/dropout1

"fast hack"
7 years ago · 12856c5f69
parent cffe1a9112 e33af2414b
commit 12856c5f69
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/operators/dropout_op.cu
+++ b/paddle/fluid/operators/dropout_op.cu
@ -33,6 +33,7 @@ __global__ void RandomGenerator(const size_t n, const int seed,

  int idx = blockDim.x * blockIdx.x + threadIdx.x;
  for (; idx < n; idx += blockDim.x * gridDim.x) {
+    rng.discard(idx);
    if (dist(rng) < dropout_prob) {
      mask_data[idx] = static_cast<T>(0);
    } else {