6 Commits (e90edce438b3751cb5ed93c00e65516b00885040)

Author	SHA1	Message	Date
Ross Wightman	f591e90b0d	Make sure num_features attr is present in vit models as with others	4 years ago
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.	4 years ago
Ross Wightman	736f209e7d	Update vision transformers to be compatible with official code. Port official ViT weights from jax impl.	4 years ago
Ross Wightman	27a93e9de7	Improve test crop for ViT models. Small now 77.85, added base weights at 79.35 top-1.	4 years ago
Ross Wightman	d4db9e7977	Add small vision transformer weights. 77.42 top-1.	4 years ago
Ross Wightman	f31933cb37	Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.	4 years ago