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.

Benjamin Bossan a5b01ec04e Add type annotations to _registry.py ... Description Add type annotations to _registry.py so that they will pass mypy --strict. Comment I was reading the code and felt that this module would be easier to understand with type annotations. Therefore, I went ahead and added the annotations. The idea with this PR is to start small to see if we can align on _how_ to annotate types. I've seen people in the past disagree on how strictly to annotate the code base, so before spending too much time on this, I wanted to check if you agree, Ross. Most of the added types should be straightforward. Some notes on the non-trivial changes: - I made no assumption about the fn passed to register_model, but maybe the type could be stricter. Are all models nn.Modules? - If I'm not mistaken, the type hint for get_arch_name was incorrect - I had to add a # type: ignore to model.__all__ = ... - I made some minor code changes to list_models to facilitate the typing. I think the changes should not affect the logic of the function. - I removed list from list(sorted(...)) because sorted returns always a list.		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	Davit update formatting and fix grad checkpointing (#7)	2 years ago
__init__.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.	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	Add a deprecation phase to module re-org	2 years ago
_features.py	Add grad_checkpointing support to features_only, test in EfficientDet.	2 years ago
_features_fx.py	Add a deprecation phase to module re-org	2 years ago
_helpers.py	Make safetensor import option for now. Improve avg/clean checkpoints ext handling a bit (more consistent).	2 years ago
_hub.py	Make safetensor import option for now. Improve avg/clean checkpoints ext handling a bit (more consistent).	2 years ago
_manipulate.py	Add a deprecation phase to module re-org	2 years ago
_pretrained.py	Add type annotations to _registry.py	2 years ago
_prune.py	Update model prune loader to use pkgutil	2 years ago
_registry.py	Add type annotations to _registry.py	2 years ago
beit.py	Davit update formatting and fix grad checkpointing (#7)	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	Davit update formatting and fix grad checkpointing (#7)	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 laion2b 320x320 ConvNeXt-Large CLIP 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
davit.py	Improve tracing of window attn models with simpler reshape logic	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	Davit update formatting and fix grad checkpointing (#7)	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	Davit update formatting and fix grad checkpointing (#7)	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	is_scripting() guard on checkpoint_seq	2 years ago
efficientformer_v2.py	Update head metadata for effformerv2	2 years ago
efficientnet.py	Add grad_checkpointing support to features_only, test in EfficientDet.	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	Improve tracing of window attn models with simpler reshape logic	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	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.	2 years 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	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	Add grad_checkpointing support to features_only, test in EfficientDet.	2 years ago
mobilevit.py	Davit update formatting and fix grad checkpointing (#7)	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	Davit update formatting and fix grad checkpointing (#7)	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	Davit update formatting and fix grad checkpointing (#7)	2 years ago
resnest.py	Davit update formatting and fix grad checkpointing (#7)	2 years ago
resnet.py	Davit update formatting and fix grad checkpointing (#7)	2 years ago
resnetv2.py	Davit update formatting and fix grad checkpointing (#7)	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	Davit update formatting and fix grad checkpointing (#7)	2 years ago
swin_transformer.py	Improve tracing of window attn models with simpler reshape logic	2 years ago
swin_transformer_v2.py	Improve tracing of window attn models with simpler reshape logic	2 years ago
swin_transformer_v2_cr.py	Improve tracing of window attn models with simpler reshape logic	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	Remove dead line	2 years ago
vision_transformer_hybrid.py	Davit update formatting and fix grad checkpointing (#7)	2 years 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	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	Davit update formatting and fix grad checkpointing (#7)	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