↓ 13 callersMethod__init__(self,ch_in,ch_out,kernel_size=3, stride=1, padding=1, groups=1, activation='relu')
src/train_acdc/lib/decoders.py:48
↓ 13 callersFunction_nfnet_cfg(
depths, channels=(256, 512, 1536, 1536), group_size=128, bottle_ratio=0.5, feat_mult=2.,
act
src/train_acdc/lib/models_timm/nfnet.py:177
↓ 12 callersFunction_mobilevit_block(d, c, s, transformer_dim, transformer_depth, patch_size=4, br=4.0)
src/train_acdc/lib/models_timm/mobilevit.py:104