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 1daa15ecc3
Initial Cait commit. Still some cleanup to do.
4 years ago
..
layers Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing. 4 years ago
pruned added efficientnet pruned weights 5 years ago
__init__.py Initial Cait commit. Still some cleanup to do. 4 years ago
byoanet.py Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing. 4 years ago
byobnet.py Update ByoaNet comments. Fix first Steam feat chs for ByobNet. 4 years ago
cait.py Initial Cait commit. Still some cleanup to do. 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 Support for huggingface hub via create_model and default_cfgs. 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 Update some comments re preliminary EfficientNet-V2 assumptions 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 torchscript compat and features_only behaviour in GhostNet PR. A few minor formatting changes. Reuse existing layers. 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
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 Support for huggingface hub via create_model and default_cfgs. 4 years ago
resnest.py Support for huggingface hub via create_model and default_cfgs. 4 years ago
resnet.py Post ResNet-RS merge cleanup. Add weight urls, adjust train/test/crop pct. 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 Support for huggingface hub via create_model and default_cfgs. 4 years ago
sknet.py Support for huggingface hub via create_model and default_cfgs. 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 A few miil weights naming tweaks to improve compat with model registry and filtering wildcards. 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