↓ 12 callersMethod__init__(self, dim, mlp_ratio=4, out_features=None, act_layer=StarReLU, drop=0., bias=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:449
↓ 11 callersFunctionconv2d_ABN(ni, nf, stride, activation="leaky_relu", kernel_size=3, activation_param=1e-2, groups=1)
mldanbooru/tresnet/tresnet.py:39
↓ 4 callersMethod_make_layer(self, block, planes, blocks, stride=1, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet.py:186
↓ 4 callersMethod_make_layer(self, block, planes, blocks, stride=1, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet_f.py:160
↓ 2 callersMethod__init__(self, layers, in_chans=3, num_classes=1000, width_factor=1.0, first_two_layers=BasicBlock)
mldanbooru/tresnet/tresnet.py:134
↓ 2 callersMethod__init__(self, layers, in_chans=3, num_classes=1000, width_factor=1.0, first_two_layers=BasicBlock)
mldanbooru/tresnet/tresnet_f.py:108
↓ 1 callersMethod__init__(self, d_model, nhead=8, dim_feedforward=2048, dropout=0.1, activation="relu",
layer_norm_eps
mldanbooru/ml_decoder/layer.py:13
Method__init__(self, encoder: MetaFormer, decoder: MSDecoder, num_queries=50, d_model=512, num_classes=1000, scale_skip=0)
mldanbooru/caformer/ml_caformer.py:14
Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
activation=nn.GELU(), normalize_bef
mldanbooru/caformer/ms_decoder.py:17
Method__init__(self, in_channels, out_channels,
kernel_size, stride=1, padding=0,
pre_norm=None, post_norm
mldanbooru/caformer/metaformer_baselines.py:209
Method__init__(self, scale_value=1.0, bias_value=0.0,
scale_learnable=True, bias_learnable=True,
mode=None,
mldanbooru/caformer/metaformer_baselines.py:257
Method__init__(self, dim, head_dim=32, num_heads=None, qkv_bias=False,
attn_drop=0., proj_drop=0., proj_bias=False,
mldanbooru/caformer/metaformer_baselines.py:276
Method__init__(self, affine_shape=None, normalized_dim=(-1, ), scale=True,
bias=True, eps=1e-5)
mldanbooru/caformer/metaformer_baselines.py:378
Method__init__(self, dim, expansion_ratio=2,
act1_layer=StarReLU, act2_layer=nn.Identity,
bias=False, kerne
mldanbooru/caformer/metaformer_baselines.py:403
Method__init__(self, dim, num_classes=1000, mlp_ratio=4, act_layer=SquaredReLU,
norm_layer=nn.LayerNorm, head_dropou
mldanbooru/caformer/metaformer_baselines.py:474
Method__init__(self, dim,
token_mixer=nn.Identity, mlp=Mlp,
norm_layer=nn.LayerNorm,
mldanbooru/caformer/metaformer_baselines.py:498
Method__init__(self, in_chans=3, num_classes=1000,
depths=[2, 2, 6, 2],
dims=[64, 128, 32
mldanbooru/caformer/metaformer_baselines.py:583
Method__init__(self, num_classes, num_of_groups=-1, decoder_embedding=768,
initial_num_features=2048, zsl=0
mldanbooru/ml_decoder/ml_decoder.py:47
Method__init__(self, d_model, nhead=8, dim_feedforward=2048, dropout=0.1, activation="relu",
layer_norm_eps
mldanbooru/ml_decoder/layer.py:58
Method__init__(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet.py:50
Method__init__(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet.py:89
Method__init__(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet_f.py:29
Method__init__(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet_f.py:68
Functionadd_ml_decoder_head(model, num_classes=-1, num_of_groups=-1, decoder_embedding=768, zsl=0, learn_query=False)
mldanbooru/ml_decoder/ml_decoder.py:11