Aman Arora
|
c7b40414bb
|
Make sure reduction ratio in resnetrs is 0.25
|
4 years ago |
|
Aman Arora
|
69f8c7123f
|
Remove hardcoded values
|
4 years ago |
|
Aman Arora
|
e7df94a760
|
Merge branch 'master' into resnet-rs
|
4 years ago |
|
Ross Wightman
|
9cc7dda6e5
|
Fixup byoanet configs to pass unit tests. Add swin_attn and swinnet26t model for testing.
|
4 years ago |
|
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.
|
4 years ago |
|
Ross Wightman
|
e5e15754c9
|
Fix coat first conv ident
|
4 years ago |
|
Ross Wightman
|
76739a7589
|
CoaT merge. Bit of formatting, fix torchscript (for non features), remove einops/einsum dep, add pretrained weight hub (url) support.
|
4 years ago |
|
Ross Wightman
|
026430c083
|
Merge branch 'master' of https://github.com/morizin/pytorch-image-models-1 into morizin-master
|
4 years ago |
|
Ross Wightman
|
a0492e3b48
|
A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.
|
4 years ago |
|
talrid
|
8c1f03e56c
|
comment
|
4 years ago |
|
talrid
|
19e1b67a84
|
old spaces
|
4 years ago |
|
talrid
|
a443865876
|
update naming and scores
|
4 years ago |
|
talrid
|
cf0e371594
|
84_0
|
4 years ago |
|
talrid
|
0968bdeca3
|
vit, tresnet and mobilenetV3 ImageNet-21K-P weights
|
4 years ago |
|
morizin
|
1e3b6d4dfc
|
Update __init__.py
|
4 years ago |
|
morizin
|
fd022fd6a2
|
Update __init__.py
|
4 years ago |
|
morizin
|
c2d5087eae
|
Add files via upload
|
4 years ago |
|
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.
|
4 years ago |
|
Ross Wightman
|
e8a64fb881
|
Test input size for efficientnet_v2s was wrong in last results run
|
4 years ago |
|
Ross Wightman
|
2df77ee5cb
|
Fix torchscript compat and features_only behaviour in GhostNet PR. A few minor formatting changes. Reuse existing layers.
|
4 years ago |
|
Ross Wightman
|
d793deb51a
|
Merge branch 'master' of https://github.com/iamhankai/pytorch-image-models into iamhankai-master
|
4 years ago |
|
Ross Wightman
|
f606c45c38
|
Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer
|
4 years ago |
|
iamhankai
|
de445e7827
|
Add GhostNet
|
4 years ago |
|
Ross Wightman
|
b3d7580df1
|
Update ByoaNet comments. Fix first Steam feat chs for ByobNet.
|
4 years ago |
|
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
|
4 years ago |
|
Ross Wightman
|
4e4b863b15
|
Missed norm.py
|
4 years ago |
|
Ross Wightman
|
7c97e66f7c
|
Remove commented code, add more consistent seed fn
|
4 years ago |
|
Ross Wightman
|
ce62f96d4d
|
ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments
|
4 years ago |
|
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.
|
4 years ago |
|
Aman Arora
|
530f8ddb84
|
Add first conv
|
4 years ago |
|
Aman Arora
|
ca2877f176
|
EOF newline
|
4 years ago |
|
Aman Arora
|
3f12ad3ebb
|
remove whitespace diff
|
4 years ago |
|
Aman Arora
|
206407a0e1
|
remove whitespace diff
|
4 years ago |
|
Aman Arora
|
b117e16128
|
Only include resnet-rs changes
|
4 years ago |
|
Aman Arora
|
844993267b
|
Add ResNet-RS models
|
4 years ago |
|
Ross Wightman
|
fb896c0b26
|
Update some comments re preliminary EfficientNet-V2 assumptions
|
4 years ago |
|
Ross Wightman
|
2b49ab7a36
|
Fix ResNetV2 pretrained classifier issue. Fixes #540
|
4 years ago |
|
Ross Wightman
|
de9dff933a
|
EfficientNet-V2S preliminary model def (for experimentation)
|
4 years ago |
|
Ross Wightman
|
2bb65bd875
|
Wrong default_cfg pool_size for L1
|
4 years ago |
|
Ross Wightman
|
bf2ca6bdf4
|
Merge jax and original weight init
|
4 years ago |
|
Ross Wightman
|
acbd698c83
|
Update README.md with updates. Small tweak to head_dist handling.
|
4 years ago |
|
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.
|
4 years ago |
|
Ross Wightman
|
c468c47a9c
|
Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size.
|
4 years ago |
|
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
|
4 years ago |
|
Ross Wightman
|
ea9c9550b2
|
Fully move ViT hybrids to their own file, including embedding module. Remove some extra DeiT models that were for benchmarking only.
|
4 years ago |
|
Ross Wightman
|
a5310a3451
|
Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update
|
4 years ago |
|
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
|
4 years ago |
|
Ross Wightman
|
a760a4c3f4
|
Some ViT cleanup, merge distilled model with main, fixup torchscript support for distilled models
|
4 years ago |
|
Ross Wightman
|
0dfc5a66bb
|
Add PiT model from https://github.com/naver-ai/pit
|
4 years ago |
|
Ross Wightman
|
51febd869b
|
Small tweak to tests for tnt model, reorder model imports.
|
4 years ago |
