pytorch-image-models/timm/models
Ross Wightman 9067be6a30 Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
..
_pruned
layers
__init__.py Better all res resolution for bulk runner 2024-08-26 22:28:01 -07:00
_builder.py exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
_efficientnet_blocks.py
_efficientnet_builder.py Add a few more test model defs in prep for weight upload 2024-09-21 11:38:38 -07:00
_factory.py
_features.py Fix #2242 add checks for out indices with intermediate getter mode 2024-07-23 08:19:09 -07:00
_features_fx.py
_helpers.py exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
_hub.py exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
_manipulate.py
_pretrained.py
_prune.py
_registry.py Better all res resolution for bulk runner 2024-08-26 22:28:01 -07:00
beit.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -07:00
byoanet.py
byobnet.py Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
cait.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
coat.py
convit.py set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers 2024-07-17 15:25:48 -07:00
convmixer.py
convnext.py Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
crossvit.py
cspnet.py
davit.py
deit.py
densenet.py Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size 2024-06-07 13:53:00 -07:00
dla.py
dpn.py
edgenext.py
efficientformer.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
efficientformer_v2.py Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
efficientnet.py Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
efficientvit_mit.py
efficientvit_msra.py
eva.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -07:00
factory.py Add a deprecation phase to module re-org 2022-12-09 14:39:45 -08:00
fastvit.py Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
features.py
focalnet.py
fx_features.py
gcvit.py
ghostnet.py
hardcorenas.py
helpers.py Add a deprecation phase to module re-org 2022-12-09 14:39:45 -08:00
hgnet.py Added gradient checkpointing to hgnet 2024-08-07 09:22:51 -07:00
hiera.py Remaining Hiera sbb weights uploaded 2024-08-21 10:06:27 -07:00
hieradet_sam2.py Add layer scale to hieradet 2024-08-21 11:23:39 -07:00
hrnet.py
hub.py
inception_next.py
inception_resnet_v2.py
inception_v3.py Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size 2024-06-07 13:53:00 -07:00
inception_v4.py
levit.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
maxxvit.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
metaformer.py Some missed reset_classifier() type annotations 2024-06-16 10:39:27 -07:00
mlp_mixer.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
mobilenetv3.py Add MobileNetV3 RA4 (mnv4 recipe) weights 2024-09-02 13:10:34 -07:00
mobilevit.py
mvitv2.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
nasnet.py
nest.py
nextvit.py Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
nfnet.py Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
pit.py
pnasnet.py
pvt_v2.py
rdnet.py Remove eval to avoid safety concerns 2024-08-07 14:54:13 -07:00
registry.py
regnet.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
repghost.py
repvit.py
res2net.py
resnest.py
resnet.py Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
resnetv2.py Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size 2024-06-07 13:53:00 -07:00
rexnet.py
selecsls.py
senet.py
sequencer.py Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size 2024-06-07 13:53:00 -07:00
sknet.py
swin_transformer.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -07:00
swin_transformer_v2.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -07:00
swin_transformer_v2_cr.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -07:00
tiny_vit.py
tnt.py
tresnet.py
twins.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
vgg.py
visformer.py Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size 2024-06-07 13:53:00 -07:00
vision_transformer.py Add weights for new tiny test models 2024-09-22 07:59:23 -07:00
vision_transformer_hybrid.py set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers 2024-07-17 15:25:48 -07:00
vision_transformer_relpos.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
vision_transformer_sam.py proj_drop wasn't being called in sam vit 2024-08-02 16:12:02 -07:00
vitamin.py set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers 2024-07-17 15:25:48 -07:00
volo.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00
vovnet.py
xception.py
xception_aligned.py
xcit.py Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set() 2024-07-22 13:33:30 -07:00