Skip to content

Commit 956d7e6

Browse files
committed
Update cvt.py
1 parent e022a47 commit 956d7e6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

timm/models/cvt.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -389,6 +389,7 @@ def __init__(
389389
mlp_ratio: float = 4.,
390390
mlp_act_layer: nn.Module = QuickGELU,
391391
use_cls_token: Tuple[bool, ...] = (False, False, True),
392+
drop_rate: float = 0.,
392393
) -> None:
393394
super().__init__()
394395
num_stages = len(dims)

0 commit comments

Comments
 (0)