↓ 8 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/backbones/vit_transoss.py:57
↓ 7 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/backbones/vit_pytorch.py:122
↓ 4 callersMethod__init__(self, in_features, out_features, s=30.0, m=0.30, easy_margin=False, ls_eps=0.0)
loss/metric_learning.py:85
↓ 2 callersMethod__init__(self, last_stride=2, block=Bottleneck,layers=[3, 4, 6, 3])
model/backbones/resnet.py:85
↓ 1 callersFunctiondo_train(cfg, model, center_criterion, train_loader, val_loader, optimizer, optimizer_center, scheduler, loss_fn, num_
processor/processor.py:89
Method__init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
model/backbones/vit_transoss.py:76
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0.,
dr
model/backbones/vit_transoss.py:105
Method__init__(self, backbone, img_size=224, feature_size=None, in_chans=3, embed_dim=768)
model/backbones/vit_transoss.py:150
Method__init__(self, img_size=224, patch_size=16, stride_size=20, in_chans=3, embed_dim=768)
model/backbones/vit_transoss.py:190
Method__init__(self, img_size=224, patch_size=16, stride_size=16, in_chans=3, num_classes=1000, embed_dim=768, depth=12,
model/backbones/vit_transoss.py:241