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

fix style

Browse Source
This commit is contained in:
PingFeng Luo 2022-01-19 17:29:21 +08:00
parent 8ba67fddd2
commit c49f1eead7
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

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