未验证 提交 1a595d8e 编写于 作者: Y Yang Zhang 提交者: GitHub

Remove `no_grad_set` usage from "conv_shift" op tests (#22683)

上级 d39d8bee
...@@ -51,12 +51,10 @@ class TestConvShiftOp(OpTest): ...@@ -51,12 +51,10 @@ class TestConvShiftOp(OpTest):
self.check_grad(['X', 'Y'], 'Out') self.check_grad(['X', 'Y'], 'Out')
def test_check_grad_ignore_x(self): def test_check_grad_ignore_x(self):
self.check_grad( self.check_grad(['Y'], 'Out')
['Y'], 'Out', max_relative_error=0.05, no_grad_set=set("X"))
def test_check_grad_ignore_y(self): def test_check_grad_ignore_y(self):
self.check_grad( self.check_grad(['X'], 'Out')
['X'], 'Out', max_relative_error=0.05, no_grad_set=set('Y'))
if __name__ == '__main__': if __name__ == '__main__':
......
...@@ -23,7 +23,6 @@ NEED_TO_FIX_OP_LIST = [ ...@@ -23,7 +23,6 @@ NEED_TO_FIX_OP_LIST = [
'affine_grid', 'affine_grid',
'backward', 'backward',
'batch_norm', 'batch_norm',
'conv_shift',
'conv2d', 'conv2d',
'conv2d_transpose', 'conv2d_transpose',
'conv3d', 'conv3d',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册