↓ 13 callersMethod__init__(self, in_planes, out_planes=19, Fch=16, scale=4, branch=2, is_aux=False, norm_layer=nn.BatchNorm2d, fmap_size
CDARTS/CDARTS_segmentation/train/seg_oprs.py:237
↓ 10 callersMethod__init__(self, in_features, hidden_features=None,
out_features=None, act_layer=nn.GELU, drop=0.)
TinyViT/models/tiny_vit.py:194
↓ 10 callersMethodbackward - Inputs grad_output: float32 (B, H, L_query, L_key) - Outputs grad_input: float32 (B, H, L_query, nu
MiniViT/Mini-DeiT/rpe_ops/rpe_index.py:42
↓ 9 callersFunction_resnet(arch, block, layers, pretrained, progress, **kwargs)
CDARTS/CDARTS_segmentation/segmentation/model/backbone/resnet.py:232