Skip to content

Commit dda4c85

Browse files
committed
Fix typo #11
1 parent 8035aa6 commit dda4c85

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

detr_tf/networks/transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ def call(self, inputs, attn_mask=None, key_padding_mask=None,
317317
attn_output_weights = tf.matmul(WQ, WK, transpose_b=True)
318318

319319
if attn_mask is not None:
320-
ann_output_weights += attn_mask
320+
attn_output_weights += attn_mask
321321

322322
"""
323323
if key_padding_mask is not None:

0 commit comments

Comments
 (0)