pytorch-image-models/timm/models
Ross Wightman b9f020a509 Allow group_size override for more efficientnet and mobilenetv3 based models 2024-08-21 16:51:38 -07:00
..
_pruned
layers
__init__.py Add WIP HieraDet impl (SAM2 backbone support) 2024-08-15 17:58:15 -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 xavier_uniform init of MNVC hybrid attention modules. Small improvement in training stability. 2024-07-26 17:03:40 -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
beit.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -07:00
byoanet.py
byobnet.py Merge pull request #2225 from huggingface/small_things 2024-07-25 20:29:13 -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 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
crossvit.py
cspnet.py
davit.py
deit.py
densenet.py
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 Allow group_size override for more efficientnet and mobilenetv3 based models 2024-08-21 16:51:38 -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
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
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
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
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 Allow group_size override for more efficientnet and mobilenetv3 based models 2024-08-21 16:51:38 -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
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 resnet50d and efficientnet_b1 ra4 (mnv4) hparam weights 2024-08-16 12:10:00 -07:00
resnetv2.py
rexnet.py
selecsls.py
senet.py
sequencer.py
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
vision_transformer.py Add remaining sbb vit betwixt/mediumd fine-tunes 2024-08-21 10:09:38 -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