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 2d8b09fe8b
Add official pretrained weights to MLP-Mixer, complete model cfgs.
4 years ago
..
layers Improved (hopefully) init for SA/SA-like layers used in ByoaNets 4 years ago
pruned added efficientnet pruned weights 5 years ago
__init__.py Initial MLP-Mixer attempt... 4 years ago
byoanet.py Improved (hopefully) init for SA/SA-like layers used in ByoaNets 4 years ago
byobnet.py Update ByoaNet comments. Fix first Steam feat chs for ByobNet. 4 years ago
coat.py Fix coat first conv ident 4 years ago
cspnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
densenet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
dla.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
dpn.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
efficientnet.py Add updated EfficientNet-V2S weights, 83.8 @ 384x384 test. Add PyTorch trained EfficientNet-B4 weights, 83.4 @ 384x384 test. Tweak non TF EfficientNet B1-B4 train/test res scaling. 4 years ago
efficientnet_blocks.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 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. 4 years ago
features.py Fix some attributions, add copyrights to some file docstrings 4 years ago
ghostnet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
gluon_resnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
gluon_xception.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
hardcorenas.py Support for huggingface hub via create_model and default_cfgs. 4 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 4 years ago
hrnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
hub.py Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example. 4 years ago
inception_resnet_v2.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
inception_v3.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
inception_v4.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
mlp_mixer.py Add official pretrained weights to MLP-Mixer, complete model cfgs. 4 years ago
mobilenetv3.py A few miil weights naming tweaks to improve compat with model registry and filtering wildcards. 4 years ago
nasnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
nfnet.py Add prelim efficientnet_v2s weights from 224x224 train, eval 83.3 @ 288. Add eca_nfnet_l1 weights, train at 256, eval 84 @ 320. 4 years ago
pit.py Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer 4 years ago
pnasnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
registry.py Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models 4 years ago
regnet.py Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size. 4 years ago
res2net.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
resnest.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
resnet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
resnetv2.py Fix ResNetV2 pretrained classifier issue. Fixes #540 4 years ago
rexnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
selecsls.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
senet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
sknet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
swin_transformer.py Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer 4 years ago
tnt.py Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer 4 years ago
tresnet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 4 years ago
vgg.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
vision_transformer.py A few miil weights naming tweaks to improve compat with model registry and filtering wildcards. 4 years ago
vision_transformer_hybrid.py Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7 4 years ago
vovnet.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
xception.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
xception_aligned.py Support for huggingface hub via create_model and default_cfgs. 4 years ago