提交 adcfde3e 编写于 作者: W wanghaoshuang

Modify unitest

上级 0c10b5ff
...@@ -50,12 +50,12 @@ class TestCTCDecodeOp(OpTest): ...@@ -50,12 +50,12 @@ class TestCTCDecodeOp(OpTest):
class TestCTCDecodeOpCase1(TestCTCDecodeOp): class TestCTCDecodeOpCase1(TestCTCDecodeOp):
def config(self): def config(self):
self.op_type = "ctc_decode" self.op_type = "ctc_decode"
self.input_lod = [[0, 11, 18]] self.input_lod = [[0, 11, 19]]
self.blank = 0 self.blank = 0
self.merge_repeated = True self.merge_repeated = True
self.input = np.array( self.input = np.array(
[0, 1, 2, 2, 0, 4, 0, 4, 5, 0, 6, 6, 0, 0, 7, 7, 7, 0]).reshape( [0, 1, 2, 2, 0, 4, 0, 4, 5, 0, 6, 6, 0, 0, 7, 7, 7, 0, 0]).reshape(
[18, 1]).astype("int32") [19, 1]).astype("int32")
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册