from local
This commit is contained in:
parent
6c6442e08a
commit
3bec4f2d30
Binary file not shown.
@ -207,9 +207,7 @@ class Transducer(nn.Module):
|
|||||||
from beam_search import greedy_search_batch, greedy_search_batch_target_input
|
from beam_search import greedy_search_batch, greedy_search_batch_target_input
|
||||||
|
|
||||||
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)
|
encoder_out, x_lens = self.encoder(x, x_lens)
|
||||||
assert torch.all(x_lens > 0)
|
|
||||||
|
|
||||||
hyps = []
|
hyps = []
|
||||||
#hyp_tokens = greedy_search_batch_target_input(self, encoder_out, x_lens, decoder_out)
|
#hyp_tokens = greedy_search_batch_target_input(self, encoder_out, x_lens, decoder_out)
|
||||||
|
|||||||
Reference in New Issue
Block a user