Skip to content

Commit 20fae68

Browse files
committed
adam op handle grad.rows().size == 0 condition
1 parent df7a266 commit 20fae68

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddle/fluid/operators/adam_op.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,6 +282,10 @@ class AdamOpKernel : public framework::OpKernel<T> {
282282
} else if (grad_var->IsType<framework::SelectedRows>()) {
283283
auto& grad =
284284
Ref(ctx.Input<framework::SelectedRows>("Grad"), "Must set Grad");
285+
if (grad.rows().size() == 0) {
286+
VLOG(3) << "grad row size is 0!!";
287+
return;
288+
}
285289
// merge duplicated rows if any.
286290
scatter::MergeAdd<DeviceContext, T> merge_func;
287291
auto grad_merge =

0 commit comments

Comments
 (0)