Update cvt.py

huggingface · Jun 8, 2024 · 956d7e6 · 956d7e6
1 parent e022a47
commit 956d7e6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/timm/models/cvt.py b/timm/models/cvt.py
@@ -389,6 +389,7 @@ def __init__(
         mlp_ratio: float = 4.,
         mlp_act_layer: nn.Module = QuickGELU,
         use_cls_token: Tuple[bool, ...] = (False, False, True),
+        drop_rate: float = 0.,
     ) -> None:
         super().__init__()
         num_stages = len(dims)