未验证 提交 649f5d74 编写于 作者: B Bai Yifan 提交者: GitHub

fix mine_hard_example bug (#12664)

上级 f54efd02
...@@ -227,6 +227,9 @@ class MineHardExamplesOp : public framework::OperatorWithKernel { ...@@ -227,6 +227,9 @@ class MineHardExamplesOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_GT( PADDLE_ENFORCE_GT(
neg_pos_ratio, 0.0f, neg_pos_ratio, 0.0f,
"neg_pos_ratio must greater than zero in max_negative mode"); "neg_pos_ratio must greater than zero in max_negative mode");
PADDLE_ENFORCE_LT(
neg_dist_threshold, 1.0f,
"neg_dist_threshold must less than one in max_negative mode");
PADDLE_ENFORCE_GT( PADDLE_ENFORCE_GT(
neg_dist_threshold, 0.0f, neg_dist_threshold, 0.0f,
"neg_dist_threshold must greater than zero in max_negative mode"); "neg_dist_threshold must greater than zero in max_negative mode");
......
...@@ -722,7 +722,7 @@ def ssd_loss(location, ...@@ -722,7 +722,7 @@ def ssd_loss(location,
}, },
attrs={ attrs={
'neg_pos_ratio': neg_pos_ratio, 'neg_pos_ratio': neg_pos_ratio,
'neg_dist_threshold': neg_pos_ratio, 'neg_dist_threshold': neg_overlap,
'mining_type': mining_type, 'mining_type': mining_type,
'sample_size': sample_size, 'sample_size': sample_size,
}) })
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册