Merge pull request #14389 from jacquesqiao/fix_sgd_op_optimize_sparse_table

sgd_op optimize selected rows do not enforce id < height
panyx0718-patch-1
Qiao Longfei 7 years ago committed by GitHub
commit 6cf8f24b1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -109,8 +109,6 @@ class SGDOpKernel : public framework::OpKernel<T> {
const auto *grad_data = grad.value().data<T>();
auto *out_data = param_out->mutable_value()->data<T>();
for (size_t i = 0; i < grad.rows().size(); i++) {
PADDLE_ENFORCE(grad.rows()[i] < grad.height(),
"Input rows index should less than height");
int64_t id_index = param_out->AutoGrownIndex(grad.rows()[i], false);
PADDLE_ENFORCE_GE(id_index, static_cast<int64_t>(0),
"id should be in the table");

Loading…
Cancel
Save