mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

typo

Browse Source
This commit is contained in:
PingFeng Luo 2022-01-19 17:22:56 +08:00
parent a4a920cc22
commit 8ba67fddd2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/wenetspeech/ASR/transducer_stateless/model.py
View File

@ -106,11 +106,14 @@ class Transducer(nn.Module):
max_sym_id = torch.max(y_padded)
assert encoder_out.size(0) == decoder_out.size(0), [
encoder_out.size(), decoder_out.size()]
encoder_out.size(),
decoder_out.size()]
assert encoder_out.size(2) == decoder_out.size(2), [
encoder_out.size(), decoder_out.size()]
encoder_out.size(),
decoder_out.size()]
assert encoder_out.size(2) >= (max_sym_id + 1), [
encoder_out.size(), max_sym_id]
encoder_out.size(),
max_sym_id]
boundary = torch.zeros(
(x.size(0), 4), dtype=torch.int64, device=x.device