↓ 4 callersMethod__init__(self, dim, input_resolution, depth, num_heads, window_size,
mlp_ratio=4., qkv_bias=True, dro
timm/models/swin_transformer.py:378
↓ 4 callersMethod_make_layer(self, block, planes, blocks, groups, reduction, stride=1,
downsample_kernel_size=1, downs
timm/models/senet.py:351
↓ 4 callersMethod_make_layer(self, block, planes, blocks, stride=1, use_se=True, aa_layer=None)
timm/models/tresnet.py:214
↓ 4 callersMethodbackward input: grad_out: (b, c, m, nsample) output: (b, c, n), None
pointops/functions/pointops.py:166
↓ 3 callersMethod__init__(self, small=False, num_init_features=64, k_r=96, groups=32,
b=False, k_sec=(3, 4, 20, 3), in
timm/models/dpn.py:169
↓ 3 callersMethod__init__(self, growth_rate=32, block_config=(6, 12, 24, 16), bn_size=4, stem_type='',
num_classes=100
timm/models/densenet.py:165
↓ 3 callersMethod__init__(self, dim, num_heads, head_dim_ratio=1., mlp_ratio=4.,
drop=0., attn_drop=0., drop_path=0.,
timm/models/visformer.py:115
↓ 3 callersMethod__init__(self, cfg, num_classes=1000, in_chans=3, drop_rate=0.0, global_pool='avg')
timm/models/selecsls.py:156
↓ 3 callersMethod__init__(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dim=768, in_dim=48, depth=12,
timm/models/tnt.py:152
↓ 3 callersMethod__init__(
self, in_channel, out_channels, kernel_size, stride=1, padding=None,
dilation=1, gro
timm/models/layers/std_conv.py:32
↓ 3 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
timm/models/layers/mlp.py:13