Skip to content

Commit 9dfa28e

Browse files
committed
[CPU-PSLIB] Modify scale_sparse_grad to scale_sparse_gradient_with_batch_size
1 parent 3518522 commit 9dfa28e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/framework/fleet/fleet_wrapper.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -940,7 +940,7 @@ void FleetWrapper::PushSparseVarsWithLabelAsync(
940940
}
941941
float* g = g_tensor->data<float>();
942942

943-
if (scale_sparse_grad && grad_dim > 0) {
943+
if (scale_sparse_gradient_with_batch_size && grad_dim > 0) {
944944
int dim = emb_dim;
945945
Eigen::Map<
946946
Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>>

0 commit comments

Comments
 (0)