Ross Wightman
|
ce69de70d3
|
Add 21k weight urls to vision_transformer. Cleanup feature_info for preact ResNetV2 (BiT) models
|
2020-12-28 16:59:15 -08:00 |
|
Ross Wightman
|
231d04e91a
|
ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing.
|
2020-12-28 16:59:15 -08:00 |
|
Ross Wightman
|
b401952caf
|
Add newly added vision transformer large/base 224x224 weights ported from JAX official repo
|
2020-10-29 17:31:01 -07:00 |
|
Ross Wightman
|
61200db0ab
|
in_chans=1 working w/ pretrained weights for vision_transformer
|
2020-10-29 15:49:36 -07:00 |
|
Ross Wightman
|
f591e90b0d
|
Make sure num_features attr is present in vit models as with others
|
2020-10-29 15:33:47 -07:00 |
|
Ross Wightman
|
f944242cb0
|
Fix #262, num_classes arg mixup. Make vision_transformers a bit closer to other models wrt get/reset classfier/forward_features. Fix torchscript for ViT.
|
2020-10-29 13:58:28 -07:00 |
|
Ross Wightman
|
736f209e7d
|
Update vision transformers to be compatible with official code. Port official ViT weights from jax impl.
|
2020-10-26 18:42:11 -07:00 |
|
Ross Wightman
|
27a93e9de7
|
Improve test crop for ViT models. Small now 77.85, added base weights at 79.35 top-1.
|
2020-10-21 23:35:25 -07:00 |
|
Ross Wightman
|
d4db9e7977
|
Add small vision transformer weights. 77.42 top-1.
|
2020-10-21 12:14:12 -07:00 |
|
Ross Wightman
|
f31933cb37
|
Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.
|
2020-10-13 13:33:44 -07:00 |