from local

This commit is contained in:
dohe0342 2023-04-27 16:34:44 +09:00
parent bff86e131a
commit 473dea7804
2 changed files with 1 additions and 1 deletions

View File

@ -139,7 +139,7 @@ class Transducer(nn.Module):
assert x.size(0) == x_lens.size(0) == y.dim0 assert x.size(0) == x_lens.size(0) == y.dim0
encoder_out, x_lens = self.encoder(x, x_lens, prompt=self.prompt) encoder_out, x_lens = self.encoder(x, x_lens, prompt=self.prompt, sid=self.sid)
assert torch.all(x_lens > 0) assert torch.all(x_lens > 0)
# compute ctc log-probs # compute ctc log-probs