def __init__()

in models/vision_transformer.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(
        self,
        image_size,
        patch_size,
        num_layers,
        num_heads,
        hidden_dim,
        mlp_dim,
        dropout_rate=0,
        attention_dropout_rate=0,
        classifier="token",
        conv_stem_layers: Union[List[ConvStemLayer], List[Dict], None] = None,
        num_classes: Optional[int] = None,