mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/models
History
Aman Arora 560eae38f5
[WIP] Add ResNet-RS models (#554) 
* Add ResNet-RS models

* Only include resnet-rs changes

* remove whitespace diff

* EOF newline

* Update time

* increase time

* Add first conv

* Try running only resnetv2_101x1_bitm on Linux runner

* Add to exclude filter

* Run test_model_forward_features for all

* Add to exclude ftrs

* back to defaults

* only run test_forward_features

* run all tests

* Run all tests

* Add bigger resnetrs to model filters to fix Github CLI

* Remove resnetv2_101x1_bitm from exclude feat features

* Remove hardcoded values

* Make sure reduction ratio in resnetrs is 0.25

* There is no bias in replaced maxpool so remove it
 		2021-05-04 10:59:44 -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 added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
__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.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
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
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 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 [WIP] Add ResNet-RS models (#554) 2021-05-04 10:59:44 -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.py A few miil weights naming tweaks to improve compat with model registry and filtering wildcards. 2021-04-28 10:33:02 -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
vovnet.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
xception_aligned.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00