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 ea9c9550b2
Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
3 years ago
..
layers Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 3 years ago
pruned added efficientnet pruned weights 4 years ago
__init__.py Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 3 years ago
byobnet.py Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments. 3 years ago
cspnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
densenet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
dla.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
dpn.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
efficientnet.py Merge pull request #502 from khatami-mehrdad/efficientnet_dg 3 years ago
efficientnet_blocks.py Move stride in EdgeResidual block to 3x3 expansion conv. Fix #414 3 years ago
efficientnet_builder.py Merge branch 'logger' into features. Change 'logger' to '_logger'. 4 years ago
factory.py A few minor fixes and bit more cleanup on the huggingface hub integration. 3 years ago
features.py Fix some attributions, add copyrights to some file docstrings 4 years ago
gluon_resnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
gluon_xception.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
hardcorenas.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
helpers.py Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433 3 years ago
hrnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
hub.py Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example. 3 years ago
inception_resnet_v2.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
inception_v3.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
inception_v4.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
mobilenetv3.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
nasnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
nfnet.py Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example. 3 years ago
pit.py Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433 3 years ago
pnasnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
registry.py Fix some attributions, add copyrights to some file docstrings 4 years ago
regnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
res2net.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
resnest.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
resnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
resnetv2.py Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 3 years ago
rexnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
selecsls.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
senet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
sknet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
tnt.py Update TNT 3 years ago
tresnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
vgg.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
vision_transformer.py Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only. 3 years ago
vision_transformer_hybrid.py Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only. 3 years ago
vovnet.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
xception.py Support for huggingface hub via create_model and default_cfgs. 3 years ago
xception_aligned.py Support for huggingface hub via create_model and default_cfgs. 3 years ago