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 e7f0db8664
Fix drop/drop_path arg on MLP-Mixer model. Fix #641
4 years ago
..
layers EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 4 years ago
pruned added efficientnet pruned weights 5 years ago
__init__.py Merge branch 'master' into cait 4 years ago
byoanet.py More adjustments to ByoaNet models for further experiments. 4 years ago
byobnet.py Update ByoaNet comments. Fix first Steam feat chs for ByobNet. 4 years ago
cait.py Fix crop_pct for cait models. 4 years ago
coat.py Rethink name of patch embed grid info 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 Change 21k model naming from _21k to _in21k for consistency with existing 21k models. 4 years ago
efficientnet_blocks.py EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 4 years ago
efficientnet_builder.py EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 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 EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 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 EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 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 Fix drop/drop_path arg on MLP-Mixer model. Fix #641 4 years ago
mobilenetv3.py EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 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 Move Mlp and PatchEmbed modules into layers. Being used in lots of models now... 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 Rethink name of patch embed grid info 4 years ago
tnt.py Move Mlp and PatchEmbed modules into layers. Being used in lots of models now... 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 Move Mlp and PatchEmbed modules into layers. Being used in lots of models now... 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