提交 67a84834 authored 作者: João Victor Tozatti Risso's avatar João Victor Tozatti Risso 提交者: João Victor Tozatti Risso

Change simple_test method to test_ctc in CPU CTC test

上级 5f2fde89
...@@ -74,7 +74,7 @@ class TestCTC(unittest.TestCase): ...@@ -74,7 +74,7 @@ class TestCTC(unittest.TestCase):
self.run_ctc(activations, labels, activation_times, expected_costs, expected_gradients) self.run_ctc(activations, labels, activation_times, expected_costs, expected_gradients)
def simple_test(self): def test_ctc(self):
activations = np.asarray([[[0.1, 0.6, 0.1, 0.1, 0.1], [0.1, 0.1, 0.6, 0.1, 0.1]], activations = np.asarray([[[0.1, 0.6, 0.1, 0.1, 0.1], [0.1, 0.1, 0.6, 0.1, 0.1]],
[[0.6, 0.1, 0.1, 0.1, 0.1], [0.1, 0.1, 0.5, 0.2, 0.1]]], [[0.6, 0.1, 0.1, 0.1, 0.1], [0.1, 0.1, 0.5, 0.2, 0.1]]],
dtype=np.float32) dtype=np.float32)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论