from local

This commit is contained in:
dohe0342 2022-12-09 17:26:03 +09:00
parent 6b3a8be81b
commit b0d160b3d4
2 changed files with 1 additions and 1 deletions

View File

@ -113,7 +113,7 @@ class Transducer(nn.Module):
lm_scale * lm_probs + am_scale * am_probs +
(1-lm_scale-am_scale) * combined_probs
"""
assert x.ndim == 3, x.shape
assert x.ndim == 2 or x.ndim == 3, x.shape
assert x_lens.ndim == 1, x_lens.shape
assert y.num_axes == 2, y.num_axes