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 08d60f4a9a resnetrs50 pool sizing wrong
2021-05-04 12:41:09 -07:00
..
layers
Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.
2021-04-29 21:08:37 -07:00
pruned
…
__init__.py
Update __init__.py
2021-04-24 18:22:59 +05:30
byoanet.py
Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.
2021-04-29 21:08:37 -07:00
byobnet.py
Update ByoaNet comments. Fix first Steam feat chs for ByobNet.
2021-04-12 12:11:35 -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
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -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
Update some comments re preliminary EfficientNet-V2 assumptions
2021-04-03 12:00:25 -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
…
ghostnet.py
Fix torchscript compat and features_only behaviour in GhostNet PR. A few minor formatting changes. Reuse existing layers.
2021-04-15 10:20:26 -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
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
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
resnest.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
resnet.py
resnetrs50 pool sizing wrong
2021-05-04 12:41:09 -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
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -07:00
sknet.py
Support for huggingface hub via create_model and default_cfgs.
2021-03-16 22:48:26 -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
A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.
2021-04-28 10:33:02 -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