未验证 提交 24ba5ee0 编写于 作者: G Guanghua Yu 提交者: GitHub

merge develop conflict (#31122)

上级 edacb629
......@@ -268,6 +268,10 @@ struct HardLabelSoftmaxWithCrossEntropyFunctor {
int64_t idx_remain = idx % remain;
// labels, loss view as [n, remain]
int64_t idx_lbl = idx_n * remain + idx_remain;
PADDLE_ENFORCE(labels_[idx_lbl] >= 0 && labels_[idx_lbl] < d_,
"The value of label[%ld] expected >= 0 and < %ld,"
"but got %ld. Please check input value.",
idx_lbl, d_, labels_[idx_lbl]);
// It also would ignore labels not in range(class_num).
if (idx_axis != labels_[idx_lbl]) {
log_softmax_[idx] = exp_on_device(log_softmax_[idx]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册