Method__init__(self, n_class=1, img_size_s1=(256,256), img_size_s2=(224,224), model_scale='small', decoder_aggregation='addi
lib/networks.py:425
Method__init__(self, n_class=1, img_size_s1=(256,256), img_size_s2=(224,224), model_scale='small', decoder_aggregation='addi
lib/networks.py:515
Method__init__(self, n_class=1, img_size_s1=(256,256), img_size_s2=(224,224), decoder_aggregation='additive', interpolation=
lib/networks.py:633
Method__init__(self, n_class=1, img_size_s1=(256,256), img_size_s2=(224,224), decoder_aggregation='additive', interpolation=
lib/networks.py:728
Method__init__(
self, in_chs, out_chs, kernel_size=1, stride=1, pad=0, dilation=1,
groups=1, bn_weig
lib/models_timm/levit.py:128
Method__init__(
self, dim, key_dim, num_heads=8, attn_ratio=4, act_layer=None, resolution=14, use_conv=False)
lib/models_timm/levit.py:241
Method__init__(
self, in_dim, out_dim, key_dim, num_heads=8, attn_ratio=2,
act_layer=None, stride=2,
lib/models_timm/levit.py:310
Method__init__(self, model, return_nodes: Union[Dict[str, str], List[str]], squeeze_out: bool = True)
lib/models_timm/fx_features.py:97
Method__init__(
self, model,
out_indices=(0, 1, 2, 3, 4), out_map=None, feature_concat=False, flatte
lib/models_timm/features.py:177
Method__init__(
self, model,
out_indices=(0, 1, 2, 3, 4), out_map=None, feature_concat=False, flatte
lib/models_timm/features.py:224
Method__init__(
self, model,
out_indices=(0, 1, 2, 3, 4), out_map=None, out_as_dict=False, no_rewrit
lib/models_timm/features.py:248
Method__init__(
self,
cfg: MaxxVitCfg,
img_size: Union[int, Tuple[int, int]] = 224,
lib/models_timm/maxxvit.py:1602
Method__init__(self, dim, num_heads=8, qkv_bias=False, rel_pos_cls=None, attn_drop=0., proj_drop=0.)
lib/models_timm/vision_transformer_relpos.py:238
Method__init__(
self, dim, num_heads, mlp_ratio=4., qkv_bias=False, rel_pos_cls=None, init_values=None,
lib/models_timm/vision_transformer_relpos.py:282
Method__init__(
self, dim, num_heads, mlp_ratio=4., qkv_bias=False, rel_pos_cls=None, init_values=None,
lib/models_timm/vision_transformer_relpos.py:306
Method__init__(
self, dim, seq_len, mlp_ratio=(0.5, 4.0), mlp_layer=Mlp,
norm_layer=partial(nn.Layer
lib/models_timm/mlp_mixer.py:150
Method__init__(
self, dim, seq_len, mlp_ratio=4, mlp_layer=Mlp, norm_layer=Affine,
act_layer=nn.GELU
lib/models_timm/mlp_mixer.py:182
Method__init__(
self, dim, seq_len, mlp_ratio=4, mlp_layer=GatedMlp,
norm_layer=partial(nn.LayerNorm
lib/models_timm/mlp_mixer.py:229
Method__init__(self, in_channels, block_size, groups, act_layer=nn.ReLU, norm_layer=nn.BatchNorm2d)
lib/models_timm/layers/non_local_attn.py:74
Method__init__(
self, in_channels, block_size=7, groups=2, rd_ratio=0.25, rd_channels=None, rd_divisor=8,
lib/models_timm/layers/non_local_attn.py:129
Method__init__(
self, in_channel, out_channels, kernel_size, stride=1, padding='SAME',
dilation=1, g
lib/models_timm/layers/std_conv.py:56
Method__init__(
self, in_channels, out_channels, kernel_size, stride=1, padding=None,
dilation=1, gr
lib/models_timm/layers/std_conv.py:85
Method__init__(
self, in_channels, out_channels, kernel_size, stride=1, padding='SAME',
dilation=1,
lib/models_timm/layers/std_conv.py:114
Method__init__(
self, dim, dim_out=None, feat_size=None, stride=1, num_heads=4, dim_head=None,
qk_ra
lib/models_timm/layers/bottleneck_attn.py:106
Method__init__(self, kernel_size: int, stride=None, padding=0, dilation=1, ceil_mode=False)
lib/models_timm/layers/pool2d_same.py:45
Method__init__(self, num_features, eps=1e-5, momentum=0.1, affine=True, apply_act=True,
act_layer="leaky_re
lib/models_timm/layers/inplace_abn.py:40
Method__init__(self, in_chs, num_classes, pool_type='avg', drop_rate=0., use_conv=False)
lib/models_timm/layers/classifier.py:41
Method__init__(
self, channels, rd_ratio=1. / 16, rd_channels=None, rd_divisor=8,
bias=True, act_lay
lib/models_timm/layers/squeeze_excite.py:86
Method__init__(self, d_model, nhead=8, dim_feedforward=2048, dropout=0.1, activation="relu",
layer_norm_eps
lib/models_timm/layers/ml_decoder.py:36
Method__init__(self, channels, use_attn=True, fuse_add=False, fuse_scale=True, init_last_zero=False,
rd_rat
lib/models_timm/layers/global_context.py:21
Method__init__(self, in_channels, out_channels, kernel_size=3, stride=1, dilation=1, padding='', bias=False,
lib/models_timm/layers/separable_conv.py:17
Method__init__(self, in_channels, out_channels, kernel_size, stride=1,
padding=0, dilation=1, groups=1, bia
lib/models_timm/layers/conv2d_same.py:24
Method__init__(
self, channels, feat_size=None, extra_params=False, extent=0, use_mlp=True,
rd_ratio
lib/models_timm/layers/gather_excite.py:28