未验证 提交 55725cd2 编写于 作者: X xiaoting 提交者: GitHub

fix for timeout, test=develop (#29788)

上级 a94c3cbb
......@@ -459,9 +459,9 @@ class TestTrilinearInterpScale1(TestTrilinearInterpOp):
def init_test_case(self):
self.interp_method = 'trilinear'
self.input_shape = [2, 3, 5, 7, 9]
self.out_d = 82
self.out_h = 60
self.out_w = 25
self.out_d = 19
self.out_h = 15
self.out_w = 8
self.scale = 2.
self.align_corners = True
self.align_mode = 1
......@@ -471,8 +471,8 @@ class TestTrilinearInterpScale2(TestTrilinearInterpOp):
def init_test_case(self):
self.interp_method = 'trilinear'
self.input_shape = [2, 3, 5, 7, 9]
self.out_d = 60
self.out_h = 40
self.out_d = 30
self.out_h = 20
self.out_w = 25
self.scale = 1.
self.align_corners = True
......@@ -483,8 +483,8 @@ class TestTrilinearInterpScale3(TestTrilinearInterpOp):
def init_test_case(self):
self.interp_method = 'trilinear'
self.input_shape = [2, 3, 5, 7, 9]
self.out_d = 60
self.out_h = 40
self.out_d = 30
self.out_h = 20
self.out_w = 25
self.scale = 1.5
self.align_corners = True
......@@ -495,8 +495,8 @@ class TestTrilinearInterpZero(TestTrilinearInterpOp):
def init_test_case(self):
self.interp_method = 'trilinear'
self.input_shape = [2, 3, 5, 7, 11]
self.out_d = 60
self.out_h = 40
self.out_d = 30
self.out_h = 20
self.out_w = 25
self.scale = 0.0
self.align_corners = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册