mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
bff86e131a
commit
473dea7804
Binary file not shown.
@ -139,7 +139,7 @@ class Transducer(nn.Module):
|
||||
|
||||
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)
|
||||
|
||||
# compute ctc log-probs
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user