Format ctc_decode test correctly.
This commit is contained in:
parent
a24a75aee1
commit
f1487900c6
@ -1771,7 +1771,10 @@ class TestCTC(test.TestCase):
|
|||||||
-3.777835 # output beam 1
|
-3.777835 # output beam 1
|
||||||
], np.float32)[np.newaxis, :]
|
], np.float32)[np.newaxis, :]
|
||||||
|
|
||||||
decode_truth = [np.array([1, 0, -1, -1, -1, -1, -1]), np.array([0, 1, 0, -1, -1, -1 ,-1])]
|
decode_truth = [
|
||||||
|
np.array([1, 0, -1, -1, -1, -1, -1]),
|
||||||
|
np.array([0, 1, 0, -1, -1, -1, -1])
|
||||||
|
]
|
||||||
beam_width = 2
|
beam_width = 2
|
||||||
top_paths = 2
|
top_paths = 2
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user