Ross Wightman
0d87650fea
Remove filter hack from BlurPool w/ non-persistent buffer. Use BlurPool2d instead of AntiAliasing.. for TResNet. Breaks PyTorch < 1.6.
2021-05-04 16:56:28 -07:00
Ross Wightman
ddc743fdf8
Update ResNet-RS models to EMA weights
2021-05-04 16:16:55 -07:00
Ross Wightman
08d60f4a9a
resnetrs50 pool sizing wrong
2021-05-04 12:41:09 -07:00
Ross Wightman
1daa15ecc3
Initial Cait commit. Still some cleanup to do.
2021-05-04 11:19:27 -07:00
Ross Wightman
67d0665b46
Post ResNet-RS merge cleanup. Add weight urls, adjust train/test/crop pct.
2021-05-04 11:04:23 -07:00
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
Ross Wightman
9cc7dda6e5
Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.
2021-04-29 21:08:37 -07:00
Ross Wightman
e15c3886ba
Defaul lambda r=7. Define '26t' stage 4/5 256x256 variants for all of bot/halo/lambda nets for experiment. Add resnet50t for exp. Fix a few comments.
2021-04-29 10:58:49 -07:00
Ross Wightman
e5e15754c9
Fix coat first conv ident
2021-04-28 18:09:23 -07:00
Ross Wightman
76739a7589
CoaT merge. Bit of formatting, fix torchscript (for non features), remove einops/einsum dep, add pretrained weight hub (url) support.
2021-04-28 16:31:35 -07:00
Ross Wightman
026430c083
Merge branch 'master' of https://github.com/morizin/pytorch-image-models-1 into morizin-master
2021-04-28 11:24:27 -07:00
Ross Wightman
a0492e3b48
A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.
2021-04-28 10:33:02 -07:00
talrid
8c1f03e56c
comment
2021-04-28 09:04:10 +03:00
talrid
19e1b67a84
old spaces
2021-04-28 09:01:29 +03:00
talrid
a443865876
update naming and scores
2021-04-28 08:57:06 +03:00
talrid
cf0e371594
84_0
2021-04-27 22:33:55 +03:00
talrid
0968bdeca3
vit, tresnet and mobilenetV3 ImageNet-21K-P weights
2021-04-27 20:54:06 +03:00
morizin
1e3b6d4dfc
Update __init__.py
2021-04-24 18:22:59 +05:30
morizin
fd022fd6a2
Update __init__.py
2021-04-24 18:22:36 +05:30
morizin
c2d5087eae
Add files via upload
2021-04-24 17:47:57 +05:30
Ross Wightman
c1cf9712fc
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
Ross Wightman
e8a64fb881
Test input size for efficientnet_v2s was wrong in last results run
2021-04-17 16:17:41 -07:00
Ross Wightman
2df77ee5cb
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
Ross Wightman
d793deb51a
Merge branch 'master' of https://github.com/iamhankai/pytorch-image-models into iamhankai-master
2021-04-15 09:30:25 -07:00
Ross Wightman
f606c45c38
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
2021-04-13 12:17:21 -07:00
iamhankai
de445e7827
Add GhostNet
2021-04-13 23:19:51 +08:00
Ross Wightman
b3d7580df1
Update ByoaNet comments. Fix first Steam feat chs for ByobNet.
2021-04-12 12:11:35 -07:00
Ross Wightman
16f7aa9f54
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
Ross Wightman
4e4b863b15
Missed norm.py
2021-04-12 09:57:56 -07:00
Ross Wightman
7c97e66f7c
Remove commented code, add more consistent seed fn
2021-04-12 09:51:36 -07:00
Ross Wightman
ce62f96d4d
ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments
2021-04-12 09:38:02 -07:00
Ross Wightman
21812d33aa
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
Ross Wightman
fb896c0b26
Update some comments re preliminary EfficientNet-V2 assumptions
2021-04-03 12:00:25 -07:00
Ross Wightman
2b49ab7a36
Fix ResNetV2 pretrained classifier issue. Fixes #540
2021-04-03 11:18:12 -07:00
Ross Wightman
de9dff933a
EfficientNet-V2S preliminary model def (for experimentation)
2021-04-02 09:36:51 -07:00
Ross Wightman
2bb65bd875
Wrong default_cfg pool_size for L1
2021-04-01 20:00:41 -07:00
Ross Wightman
bf2ca6bdf4
Merge jax and original weight init
2021-04-01 18:11:51 -07:00
Ross Wightman
acbd698c83
Update README.md with updates. Small tweak to head_dist handling.
2021-04-01 17:49:05 -07:00
Ross Wightman
9071568f0e
Add weights for SE NFNet-L0 model, rename nfnet_l0b -> nfnet_l0. 82.75 top-1 @ 288. Add nfnet_l1 model def for training.
2021-04-01 17:22:27 -07:00
Ross Wightman
c468c47a9c
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
Ross Wightman
288682796f
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
Ross Wightman
ea9c9550b2
Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
2021-04-01 14:17:38 -07:00
Ross Wightman
a5310a3451
Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
2021-04-01 12:15:34 -07:00
Ross Wightman
7953e5d11a
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
Ross Wightman
a760a4c3f4
Some ViT cleanup, merge distilled model with main, fixup torchscript support for distilled models
2021-03-31 18:21:02 -07:00
Ross Wightman
0dfc5a66bb
Add PiT model from https://github.com/naver-ai/pit
2021-03-31 18:20:14 -07:00
Ross Wightman
51febd869b
Small tweak to tests for tnt model, reorder model imports.
2021-03-29 11:33:08 -07:00
contrastive
de86314655
Update TNT
2021-03-29 08:23:34 +08:00
contrastive
cfc15283a4
Update TNT url
2021-03-28 23:19:15 +08:00
contrastive
4a09bc851e
Add TNT model
2021-03-28 19:53:42 +08:00
