Method__init__(self, in_dim, hidden_dim, out_chans=None, act_layer=nn.GELU, dropout=0.)
modules/VIPTRv1-T_ch.py:316
Method__init__(self, retention: str, embed_dim: int, num_heads: int, ffn_dim: int, out_dim, drop_path=0.,
l
modules/VIPTRv1-T_ch.py:516
Method__init__(self, dim, resolution, idx, split_size=7, dim_out=None, num_heads=8, attn_drop=0., proj_drop=0.,
modules/VIPTRv1-T_ch.py:588
Method__init__(self, dim, reso, num_heads,
split_size=7, mlp_ratio=4., qkv_bias=False, qk_scale=None,
modules/VIPTRv1-T_ch.py:674
Method__init__(self, embed_dim, out_dim, depth, num_heads,
init_value: float, heads_range: float, mlp_ratio
modules/VIPTRv1-T_ch.py:754
Method__init__(self, in_chans=3, out_dim=192,
embed_dims=[96, 192, 384, 768], depths=[2, 2, 6, 2], num_head
modules/VIPTRv1-T_ch.py:882
Method__init__(self, retention: str, embed_dim: int, num_heads: int, ffn_dim: int, out_dim, drop_path=0., layerscale=False,
modules/VIPTRv2T_ch.py:447
Method__init__(self, dim, resolution, idx, split_size=7, dim_out=None, num_heads=8, attn_drop=0., proj_drop=0.,
modules/VIPTRv2T_ch.py:516
Method__init__(self, dim, reso, num_heads,
split_size=7, mlp_ratio=4., qkv_bias=False, qk_scale=None,
modules/VIPTRv2T_ch.py:603
Method__init__(self, embed_dim, out_dim, depth, num_heads,
init_value: float, heads_range: float, mlp_rati
modules/VIPTRv2T_ch.py:683
Method__init__(self, in_chans=3, out_dim=192,
embed_dims=[96, 192, 384, 768], depths=[2, 2, 6, 2], num_hea
modules/VIPTRv2T_ch.py:770