refine code test=develop

fe3995d3 · Qiao Longfei · 56686d0f · fe3995d3
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/operators/optimizers/adam_op.h paddle/fluid/operators/optimizers/adam_op.h +3 -1

未找到文件。
--- a/paddle/fluid/operators/optimizers/adam_op.h
+++ b/paddle/fluid/operators/optimizers/adam_op.h
@@ -227,7 +227,9 @@ struct SparseAdamFunctor {
  inline HOSTDEVICE void operator()(size_t i) const {
    auto row_idx =
        math::BinarySearch<int64_t>(rows_, row_count_, i / row_numel_);
-    if (!(lazy_mode_ && row_idx < 0)) {
+    if (lazy_mode_ && row_idx < 0) {
+      return;
+    } else {
      T g = row_idx >= 0 ? grad_[row_idx * row_numel_ + i % row_numel_] : 0;
      adam_update(i, g);
    }