We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bc1085b commit cde3a7dCopy full SHA for cde3a7d
timm/models/vision_transformer.py
@@ -300,7 +300,9 @@ def __init__(
300
drop_path: float = 0.,
301
act_layer: Type[nn.Module] = nn.GELU,
302
norm_layer: Type[nn.Module] = LayerNorm,
303
- mlp_layer: Optional[Type[nn.Module]] = None,
+ mlp_layer: Optional[Type[nn.Module]] = None, # not used
304
+ attn_layer: Optional[LayerType] = None, # not used
305
+ depth: int = 0, # not used
306
device = None,
307
dtype = None,
308
) -> None:
0 commit comments