Skip to content

Commit b927237

Browse files
committed
Update ml_decoder.py
1 parent abbdf05 commit b927237

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: timm/layers/ml_decoder.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ def __init__(
349349
learnable_class_embed: bool = False,
350350
embed_drop: float = 0.1,
351351
embed_norm: bool = True,
352-
k_norm: bool = False,
352+
qk_norm: bool = False,
353353
attn_drop: float = 0.1,
354354
mlp_ratio: float = 8/3,
355355
proj_drop: float = 0.1,

0 commit comments

Comments
 (0)