未验证 提交 5dbafe38 编写于 作者: L Liufang Sang 提交者: GitHub

fix smooth l1 loss op shape > 100 test=develop (#22605)

* fix smooth l1 loss op shape > 100 test=develop

* remove from white list test=develop
上级 433cef03
......@@ -64,7 +64,7 @@ class TestSmoothL1LossOp1(OpTest):
class TestSmoothL1LossOp2(OpTest):
def setUp(self):
self.op_type = "smooth_l1_loss"
dims = (5, 10)
dims = (5, 20)
self.inputs = {
'X': np.random.random(dims).astype("float32"),
'Y': np.random.random(dims).astype("float32"),
......@@ -88,7 +88,7 @@ class TestSmoothL1LossOp2(OpTest):
self.check_output()
def test_check_grad_normal(self):
self.check_grad(['X', 'Y'], 'Out')
self.check_grad(['X', 'Y'], 'Out', max_relative_error=0.03)
def test_check_grad_ingore_x(self):
self.check_grad(
......
......@@ -26,7 +26,6 @@ NEED_TO_FIX_OP_LIST = [
'matmul',
'mul',
'scatter',
'smooth_l1_loss',
'soft_relu',
'squared_l2_distance',
'tree_conv',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册