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
nateraw 14b84e8895
📝 update docstrings
2 years ago
..
_pruned Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
layers Add FlexiViT models and weights, refactoring, push more weights 2 years ago
__init__.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
_builder.py Change order of checkpoitn filtering fn application in builder, try dict, model variant first 2 years ago
_efficientnet_blocks.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
_efficientnet_builder.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
_factory.py 📝 update docstrings 2 years ago
_features.py Add a deprecation phase to module re-org 2 years ago
_features_fx.py Add a deprecation phase to module re-org 2 years ago
_helpers.py Add a deprecation phase to module re-org 2 years ago
_hub.py Factor out readme generation in hub helper, add more readme fields 2 years ago
_manipulate.py Add a deprecation phase to module re-org 2 years ago
_pretrained.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
_prune.py Add a deprecation phase to module re-org 2 years ago
_registry.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
beit.py Switch BEiT to HF hub weights 2 years ago
byoanet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
byobnet.py Pass through --model-kwargs (and --opt-kwargs for train) from command line through to model __init__. Update some models to improve arg overlay. Cleanup along the way. 2 years ago
cait.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
coat.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
convit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
convmixer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
convnext.py Add convnext_tiny and convnext_small 384x384 fine-tunes of in12k weights, fix pool size for laion CLIP convnext weights 2 years ago
crossvit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
cspnet.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 2 years ago
deit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
densenet.py More kwarg handling tweaks, maxvit_base_rw def added 2 years ago
dla.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
dpn.py More kwarg handling tweaks, maxvit_base_rw def added 2 years ago
edgenext.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
efficientformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
efficientnet.py Update efficientnet.py and convnext.py to multi-weight, add ImageNet-12k pretrained EfficientNet-B5 and ConvNeXt-Nano. 2 years ago
factory.py Add a deprecation phase to module re-org 2 years ago
features.py Add a deprecation phase to module re-org 2 years ago
fx_features.py Add a deprecation phase to module re-org 2 years ago
gcvit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
ghostnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
gluon_resnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
gluon_xception.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
hardcorenas.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
helpers.py Add a deprecation phase to module re-org 2 years ago
hrnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
hub.py Add a deprecation phase to module re-org 2 years ago
inception_resnet_v2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
inception_v3.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
inception_v4.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
levit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
maxxvit.py Push all MaxxViT weights to HF hub, cleanup impl, add feature map extraction support and prompote to 'std' architecture. Fix norm head for proper embedding / feat map output. Add new in12k + ft 1k weights. 2 years ago
mlp_mixer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
mobilenetv3.py Convert mobilenetv3 to multi-weight, tweak PretrainedCfg metadata 2 years ago
mobilevit.py More kwarg handling tweaks, maxvit_base_rw def added 2 years ago
mvitv2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
nasnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
nest.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
nfnet.py Pass through --model-kwargs (and --opt-kwargs for train) from command line through to model __init__. Update some models to improve arg overlay. Cleanup along the way. 2 years ago
pit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
pnasnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
poolformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
pvt_v2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
registry.py Add a deprecation phase to module re-org 2 years ago
regnet.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 2 years ago
res2net.py Improving kwarg merging in more models 2 years ago
resnest.py Improving kwarg merging in more models 2 years ago
resnet.py More kwarg handling tweaks, maxvit_base_rw def added 2 years ago
resnetv2.py More kwarg handling tweaks, maxvit_base_rw def added 2 years ago
rexnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
selecsls.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
senet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
sequencer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
sknet.py Fix ResNet based models to work w/ norm layers w/o affine params. Reformat long arg lists into vertical form. 2 years ago
swin_transformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
swin_transformer_v2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
swin_transformer_v2_cr.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
tnt.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
tresnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
twins.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
vgg.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
visformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
vision_transformer.py Re-order vit pretrained entries for more sensible default weights (no .tag specified) 2 years ago
vision_transformer_hybrid.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
vision_transformer_relpos.py Add FlexiViT models and weights, refactoring, push more weights 2 years ago
volo.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
vovnet.py Pass through --model-kwargs (and --opt-kwargs for train) from command line through to model __init__. Update some models to improve arg overlay. Cleanup along the way. 2 years ago
xception.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago
xception_aligned.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 2 years ago
xcit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 2 years ago