未验证 提交 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): ...@@ -64,7 +64,7 @@ class TestSmoothL1LossOp1(OpTest):
class TestSmoothL1LossOp2(OpTest): class TestSmoothL1LossOp2(OpTest):
def setUp(self): def setUp(self):
self.op_type = "smooth_l1_loss" self.op_type = "smooth_l1_loss"
dims = (5, 10) dims = (5, 20)
self.inputs = { self.inputs = {
'X': np.random.random(dims).astype("float32"), 'X': np.random.random(dims).astype("float32"),
'Y': np.random.random(dims).astype("float32"), 'Y': np.random.random(dims).astype("float32"),
...@@ -88,7 +88,7 @@ class TestSmoothL1LossOp2(OpTest): ...@@ -88,7 +88,7 @@ class TestSmoothL1LossOp2(OpTest):
self.check_output() self.check_output()
def test_check_grad_normal(self): 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): def test_check_grad_ingore_x(self):
self.check_grad( self.check_grad(
......
...@@ -26,7 +26,6 @@ NEED_TO_FIX_OP_LIST = [ ...@@ -26,7 +26,6 @@ NEED_TO_FIX_OP_LIST = [
'matmul', 'matmul',
'mul', 'mul',
'scatter', 'scatter',
'smooth_l1_loss',
'soft_relu', 'soft_relu',
'squared_l2_distance', 'squared_l2_distance',
'tree_conv', 'tree_conv',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册