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

fix for timeout, test=develop (#29788)

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