from local

This commit is contained in:
dohe0342 2023-04-11 16:37:42 +09:00
parent deb48a94a3
commit e9e96009f5
2 changed files with 1 additions and 1 deletions

View File

@ -119,7 +119,7 @@ class Transducer(nn.Module):
lm_scale * lm_probs + am_scale * am_probs + lm_scale * lm_probs + am_scale * am_probs +
(1-lm_scale-am_scale) * combined_probs (1-lm_scale-am_scale) * combined_probs
""" """
print(self.prompt.grad) print(self.prompt[0][0])
assert x.ndim == 2 or x.ndim == 3, x.shape assert x.ndim == 2 or x.ndim == 3, x.shape
assert x_lens.ndim == 1, x_lens.shape assert x_lens.ndim == 1, x_lens.shape
assert y.num_axes == 2, y.num_axes assert y.num_axes == 2, y.num_axes