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

Update decoder.py

Browse Source
This commit is contained in:
Fangjun Kuang 2021-12-22 23:49:31 +08:00 committed by GitHub
parent c24b610389
commit 834114f1c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/transducer_stateless/decoder.py
View File

@ -76,8 +76,8 @@ class Decoder(nn.Module):
y: y:
A 2-D tensor of shape (N, U) with blank prepended. A 2-D tensor of shape (N, U) with blank prepended.
need_pad: need_pad:
True to left padding the input. Should be True during training. True to left pad the input. Should be True during training.
False to not pad the input. Used only during inference. False to not pad the input. Should be False during inference.
Returns: Returns:
Return a tensor of shape (N, U, embedding_dim). Return a tensor of shape (N, U, embedding_dim).
""" """