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 7629d8264d
Add two new SE-ResNeXt101-D 32x8d weights, one anti-aliased and one not. Reshuffle default_cfgs vs model entrypoints for resnet.py so they are better aligned.
3 years ago
..
layers Make a few more layers symbolically traceable (remove from FX leaf modules) 3 years ago
pruned
__init__.py Merge remote-tracking branch 'origin/master' into norm_norm_norm 3 years ago
beit.py Significant model refactor and additions: 3 years ago
byoanet.py Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks. 3 years ago
byobnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
cait.py Significant model refactor and additions: 3 years ago
coat.py Significant model refactor and additions: 3 years ago
convit.py Significant model refactor and additions: 3 years ago
convmixer.py Significant model refactor and additions: 3 years ago
convnext.py fix convnext args 3 years ago
crossvit.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
cspnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
deit.py Fix pit implementation to be clsoer to deit/levit re distillation head handling 3 years ago
densenet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
dla.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
dpn.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
efficientnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
efficientnet_blocks.py Merge remote-tracking branch 'origin/master' into norm_norm_norm 3 years ago
efficientnet_builder.py Merge remote-tracking branch 'origin/master' into norm_norm_norm 3 years ago
factory.py Revert ml-decoder changes to model factory and train script 3 years ago
features.py Significant model refactor and additions: 3 years ago
fx_features.py Make a few more layers symbolically traceable (remove from FX leaf modules) 3 years ago
ghostnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
gluon_resnet.py Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks. 3 years ago
gluon_xception.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
hardcorenas.py Significant model refactor and additions: 3 years ago
helpers.py Add initial swinv2 tiny / small weights 3 years ago
hrnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
hub.py Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks. 3 years ago
inception_resnet_v2.py Significant model refactor and additions: 3 years ago
inception_v3.py Significant model refactor and additions: 3 years ago
inception_v4.py Significant model refactor and additions: 3 years ago
levit.py Fix pit implementation to be clsoer to deit/levit re distillation head handling 3 years ago
mlp_mixer.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
mobilenetv3.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
mobilevit.py Significant model refactor and additions: 3 years ago
nasnet.py Significant model refactor and additions: 3 years ago
nest.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
nfnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
pit.py Fix pit implementation to be clsoer to deit/levit re distillation head handling 3 years ago
pnasnet.py Significant model refactor and additions: 3 years ago
poolformer.py Significant model refactor and additions: 3 years ago
registry.py Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks. 3 years ago
regnet.py Update test crop for new RegNet-V weights to match Y 3 years ago
res2net.py Significant model refactor and additions: 3 years ago
resnest.py Significant model refactor and additions: 3 years ago
resnet.py Add two new SE-ResNeXt101-D 32x8d weights, one anti-aliased and one not. Reshuffle default_cfgs vs model entrypoints for resnet.py so they are better aligned. 3 years ago
resnetv2.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
rexnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
selecsls.py Significant model refactor and additions: 3 years ago
senet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
sknet.py Significant model refactor and additions: 3 years ago
swin_transformer.py Add initial swinv2 tiny / small weights 3 years ago
swin_transformer_v2_cr.py Add initial swinv2 tiny / small weights 3 years ago
tnt.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
tresnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
twins.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
vgg.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
visformer.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
vision_transformer.py Add layer scale and parallel blocks to vision_transformer 3 years ago
vision_transformer_hybrid.py Merge remote-tracking branch 'origin/master' into norm_norm_norm 3 years ago
volo.py Significant model refactor and additions: 3 years ago
vovnet.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
xception.py Significant model refactor and additions: 3 years ago
xception_aligned.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago
xcit.py Fix correctness of some group matching regex (no impact on result), some formatting, missed forward_head for resnet 3 years ago