from local

This commit is contained in:
dohe0342 2023-04-11 16:34:14 +09:00
parent 91c2816a38
commit 8871dd53d2
2 changed files with 1 additions and 0 deletions

View File

@ -118,6 +118,7 @@ class Transducer(nn.Module):
lm_scale * lm_probs + am_scale * am_probs +
(1-lm_scale-am_scale) * combined_probs
"""
print(self.prompt[0][0])
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