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 cf324ea38f
Fix grad checkpointing in focalnet
1 year ago
..
_pruned Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
layers Davit update formatting and fix grad checkpointing (#7) 1 year ago
__init__.py Merge branch 'main' into focalnet_and_swin_refactor 1 year ago
_builder.py Change order of checkpoitn filtering fn application in builder, try dict, model variant first 1 year ago
_efficientnet_blocks.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
_efficientnet_builder.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
_factory.py Add a deprecation phase to module re-org 1 year ago
_features.py Add grad_checkpointing support to features_only, test in EfficientDet. 1 year ago
_features_fx.py Add a deprecation phase to module re-org 1 year ago
_helpers.py Add support to load safetensors weights 1 year ago
_hub.py changed warning to info 1 year ago
_manipulate.py Add a deprecation phase to module re-org 1 year ago
_pretrained.py Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg 1 year ago
_prune.py Update model prune loader to use pkgutil 1 year ago
_registry.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
beit.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
byoanet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
byobnet.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
cait.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
coat.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
convit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
convmixer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
convnext.py Add 384x384 convnext_large_mlp laion2b fine-tune on in1k 1 year ago
crossvit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
cspnet.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 1 year ago
davit.py Factor NormMlpClassifierHead from MaxxViT and use across MaxxViT / ConvNeXt / DaViT, refactor some type hints & comments 1 year ago
deit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
densenet.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
dla.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
dpn.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
edgenext.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
efficientformer.py is_scripting() guard on checkpoint_seq 1 year ago
efficientformer_v2.py Update head metadata for effformerv2 1 year ago
efficientnet.py Add grad_checkpointing support to features_only, test in EfficientDet. 1 year ago
factory.py Add a deprecation phase to module re-org 1 year ago
features.py Add a deprecation phase to module re-org 1 year ago
focalnet.py Fix grad checkpointing in focalnet 1 year ago
fx_features.py Add a deprecation phase to module re-org 1 year ago
gcvit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
ghostnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
gluon_resnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
gluon_xception.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
hardcorenas.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
helpers.py Add a deprecation phase to module re-org 1 year ago
hrnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
hub.py Add a deprecation phase to module re-org 1 year ago
inception_resnet_v2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
inception_v3.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
inception_v4.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
levit.py Add EfficientFormer-V2, refactor EfficientFormer and Levit for more uniformity across the 3 related arch. Add features_out support to levit conv models and efficientformer_v2. All weights on hub. 1 year ago
maxxvit.py Add Final annotation to attn_fas to avoid symbol lookup of new scaled_dot_product_attn fn on old PyTorch in jit 1 year ago
mlp_mixer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
mobilenetv3.py Add grad_checkpointing support to features_only, test in EfficientDet. 1 year ago
mobilevit.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
mvitv2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
nasnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
nest.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
nfnet.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
pit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
pnasnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
poolformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
pvt_v2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
registry.py Add a deprecation phase to module re-org 1 year ago
regnet.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 1 year ago
res2net.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
resnest.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
resnet.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
resnetv2.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
rexnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
selecsls.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
senet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
sequencer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
sknet.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
swin_transformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
swin_transformer_v2.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year 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 1 year ago
tnt.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
tresnet.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
twins.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
vgg.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
visformer.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
vision_transformer.py Remove dead line 1 year ago
vision_transformer_hybrid.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
vision_transformer_relpos.py Add Final annotation to attn_fas to avoid symbol lookup of new scaled_dot_product_attn fn on old PyTorch in jit 1 year ago
volo.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
vovnet.py Davit update formatting and fix grad checkpointing (#7) 1 year ago
xception.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago
xception_aligned.py Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 1 year ago
xcit.py Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models 1 year ago