Skip to content

Commit 5770d0e

Browse files
committed
better comment
1 parent 5ee8c35 commit 5770d0e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

i6_models/decoder/attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def forward(
4343
:param value: encoder values of shape [B,T,D_v]
4444
:param query: query of shape [B,D_k]
4545
:param weight_feedback: shape is [B,T,D_k]
46-
:param enc_seq_len: [B]
46+
:param enc_seq_len: encoder sequence lengths [B]
4747
:return: attention context [B,D_v], attention weights [B,T,1]
4848
"""
4949

0 commit comments

Comments
 (0)