You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/models
Ross Wightman 7c2660576d
Tweak init for convnext block using maxxvit/coatnext.
2 years ago
..
layers MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies.. 2 years ago
pruned
__init__.py MaxxVit! A very configurable MaxVit and CoAtNet impl with lots of goodies.. 2 years ago
beit.py Remove some redundant requires_grad=True from nn.Parameter in third party code 2 years ago
byoanet.py
byobnet.py
cait.py Remove some redundant requires_grad=True from nn.Parameter in third party code 2 years ago
coat.py
convit.py
convmixer.py
convnext.py For ConvNeXt, use timm internal LayerNorm for fast_norm in non conv_mlp mode 2 years ago
crossvit.py
cspnet.py Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks. 2 years ago
deit.py Add DeiT-III 'medium' model defs and weights 2 years ago
densenet.py Fix two default args in DenseNet blocks... fix #1427 2 years ago
dla.py
dpn.py
edgenext.py Add edgenext_base model def & weight link, update to improve ONNX export #1385 2 years ago
efficientformer.py More test fixes, pool size for 256x256 maxvit models 2 years ago
efficientnet.py
efficientnet_blocks.py
efficientnet_builder.py
factory.py
features.py
fx_features.py Update type hint for `register_notrace_module` 2 years ago
gcvit.py More model and test fixes 2 years ago
ghostnet.py
gluon_resnet.py
gluon_xception.py
hardcorenas.py
helpers.py
hrnet.py
hub.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
maxxvit.py Tweak init for convnext block using maxxvit/coatnext. 2 years ago
mlp_mixer.py Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355 2 years ago
mobilenetv3.py Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355 2 years ago
mobilevit.py Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355 2 years ago
mvitv2.py More model and test fixes 2 years ago
nasnet.py
nest.py
nfnet.py
pit.py
pnasnet.py
poolformer.py Remove some redundant requires_grad=True from nn.Parameter in third party code 2 years ago
pvt_v2.py More model and test fixes 2 years ago
registry.py
regnet.py
res2net.py
resnest.py
resnet.py
resnetv2.py
rexnet.py
selecsls.py
senet.py
sequencer.py
sknet.py
swin_transformer.py
swin_transformer_v2.py
swin_transformer_v2_cr.py
tnt.py
tresnet.py Change models with mean 0,0,0 std 1,1,1 from int to float for consistency as mentioned in #1355 2 years ago
twins.py
vgg.py
visformer.py
vision_transformer.py Take `no_emb_class` into account when calling `resize_pos_embed` 2 years ago
vision_transformer_hybrid.py
vision_transformer_relpos.py
volo.py
vovnet.py
xception.py
xception_aligned.py
xcit.py Remove some redundant requires_grad=True from nn.Parameter in third party code 2 years ago