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 89f26b9 commit d66283dCopy full SHA for d66283d
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