提交 d053561d 编写于 作者: Z zhupengyang 提交者: Tao Luo

use large input size in tree_conv and teacher_student_sigmoid_loss (#21896)

test=develop
上级 46f9184a
...@@ -28,7 +28,7 @@ class TestTeacherStudentSigmoidLossOp(OpTest): ...@@ -28,7 +28,7 @@ class TestTeacherStudentSigmoidLossOp(OpTest):
def setUp(self): def setUp(self):
self.op_type = "teacher_student_sigmoid_loss" self.op_type = "teacher_student_sigmoid_loss"
batch_size = 16 batch_size = 100
num_classes = 1 num_classes = 1
self.inputs = { self.inputs = {
'X': logit( 'X': logit(
......
...@@ -54,7 +54,7 @@ class TestTreeConvOp(OpTest): ...@@ -54,7 +54,7 @@ class TestTreeConvOp(OpTest):
self.fea_size = 3 self.fea_size = 3
self.output_size = 1 self.output_size = 1
self.max_depth = 2 self.max_depth = 2
self.batch_size = 1 self.batch_size = 2
self.num_filters = 1 self.num_filters = 1
adj_array = [ adj_array = [
1, 2, 1, 3, 1, 4, 1, 5, 2, 6, 2, 7, 2, 8, 4, 9, 4, 10, 5, 11, 6, 12, 1, 2, 1, 3, 1, 4, 1, 5, 2, 6, 2, 7, 2, 8, 4, 9, 4, 10, 5, 11, 6, 12,
......
...@@ -20,6 +20,5 @@ NOT_CHECK_OP_LIST = [ ...@@ -20,6 +20,5 @@ NOT_CHECK_OP_LIST = [
NEED_TO_FIX_OP_LIST = [ NEED_TO_FIX_OP_LIST = [
'sequence_scatter', 'log_loss', 'sequence_topk_avg_pooling', 'matmul', 'sequence_scatter', 'log_loss', 'sequence_topk_avg_pooling', 'matmul',
'add_position_encoding', 'margin_rank_loss', 'elementwise_pow', 'add_position_encoding', 'margin_rank_loss', 'elementwise_pow',
'fused_elemwise_activation', 'tree_conv', 'mul', 'fused_elemwise_activation', 'mul'
'teacher_student_sigmoid_loss'
] ]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册