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
revert-22710-feature/integrated_ps_api
Liufang Sang 5 years ago committed by GitHub
parent 433cef03e5
commit 5dbafe3805
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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',

Loading…
Cancel
Save