"fast hack"

e33af241 · dzhwinter · ab5ecdf6 · e33af241
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/dropout_op.cu paddle/fluid/operators/dropout_op.cu +1 -0

未找到文件。
--- 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 {