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:20:28 +08:00
parent 356056cea2
commit a4a920cc22
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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