mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/models
History
Ross Wightman 5fcddb96a8 Merge branch 'master' into cait
2021-05-05 17:29:38 -07:00
..
layers
Improved (hopefully) init for SA/SA-like layers used in ByoaNets
2021-05-04 21:40:39 -07:00
pruned
added efficientnet pruned weights
2020-05-03 09:29:45 +03:00
__init__.py
Merge branch 'master' into cait
2021-05-05 17:29:38 -07:00
byoanet.py
Improved (hopefully) init for SA/SA-like layers used in ByoaNets
2021-05-04 21:40:39 -07:00
byobnet.py
Update ByoaNet comments. Fix first Steam feat chs for ByobNet.
2021-04-12 12:11:35 -07:00
cait.py
Finish CaiT cleanup
2021-05-05 17:28:19 -07:00
coat.py
Fix coat first conv ident
2021-04-28 18:09:23 -07:00
cspnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
densenet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
dla.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
dpn.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
efficientnet_blocks.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
efficientnet_builder.py
Merge branch 'logger' into features. Change 'logger' to '_logger'.
2020-07-27 18:00:46 -07:00
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.
2021-04-19 10:42:56 -07:00
factory.py
A few minor fixes and bit more cleanup on the huggingface hub integration.
2021-03-17 13:18:52 -07:00
features.py
Fix some attributions, add copyrights to some file docstrings
2020-07-27 13:44:56 -07:00
ghostnet.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
gluon_resnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
gluon_xception.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
hardcorenas.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
helpers.py
Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433
2021-03-31 23:11:28 -07:00
hrnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
hub.py
Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example.
2021-03-18 11:38:06 -07:00
inception_resnet_v2.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
inception_v3.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
inception_v4.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
mlp_mixer.py
Add official pretrained weights to MLP-Mixer, complete model cfgs.
2021-05-05 15:59:40 -07:00
mobilenetv3.py
A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.
2021-04-28 10:33:02 -07:00
nasnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
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.
2021-04-11 23:26:13 -07:00
pit.py
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
2021-04-13 12:17:21 -07:00
pnasnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
registry.py
Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
2021-04-12 11:54:22 -07:00
regnet.py
Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size.
2021-04-01 16:41:04 -07:00
res2net.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
resnest.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
resnet.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
resnetv2.py
Fix ResNetV2 pretrained classifier issue. Fixes #540
2021-04-03 11:18:12 -07:00
rexnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
selecsls.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
senet.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
sknet.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
swin_transformer.py
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
2021-04-13 12:17:21 -07:00
tnt.py
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
2021-04-13 12:17:21 -07:00
tresnet.py
Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
2021-05-04 21:27:15 -07:00
vgg.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
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
2021-04-01 16:40:12 -07:00
vision_transformer.py
A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.
2021-04-28 10:33:02 -07:00
vovnet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
xception_aligned.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
xception.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00