def __init__()

in models/swin_transformer_3d.py [0:0]

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

    def __init__(
        self,
        pretrained=None,
        pretrained2d=True,
        pretrained3d=None,
        pretrained_model_key="base_model",
        patch_size=(4, 4, 4),
        in_chans=3,
        embed_dim=96,
        depths=[2, 2, 6, 2],
        num_heads=[3, 6, 12, 24],
        window_size=(2, 7, 7),
        mlp_ratio=4.0,
        qkv_bias=True,
        qk_scale=None,
        drop_rate=0.0,
        attn_drop_rate=0.0,
        drop_path_rate=0.2,
        norm_layer=nn.LayerNorm,
        patch_norm=False,
        frozen_stages=-1,
        depth_mode=None,
        depth_patch_embed_separate_params=True,