timm / image-classification Open weights updated 1 year ago

vit_base_patch16_224.augreg2_in21k_ft_in1k

Name: vit_base_patch16_224.augreg2_in21k_ft_in1k
Author: timm

A Vision Transformer (ViT) image classification model. Trained on ImageNet-21k by paper authors and (re) fine-tuned on ImageNet-1k with additional augmentation and regularization by Ross Wightman.

Params

90 M

Context

—

Downloads 30d

402 K

Likes

Commercial use: allowed apache-2.0 Not gated SAFETENSORS View on Hugging Face ↗

Download history

daily snapshots · 106 days

▲ 58 K in the last 30 days (12.5%)

538 K398 K

Jul 28Sep 1Oct 6Nov 10

Can you run it?

Estimated VRAM at 8K context unless noted. Pick your hardware to see the verdict per quantization.

File	Quant	Size	Est. VRAM	Verdict on RTX 4090 · 24 GB
model.safetensors	f32	0.3 GB	0.9 GB	✅ Runs comfortably

Estimate: file size × 1.1 + KV cache at 8K + 0.5 GB overhead. Not a benchmark — how we calculate this.

Specifications

Parameters: 90 M
Tensor type: F32
Licence: apache-2.0
First seen on the Hub: 2022-12-22
Training datasets: imagenet-1k, imagenet-21k
Added to our catalog: 2026-07-28

Compare with any image-classification model

Popular comparisons

vit-base-patch16-224 15.0 M Gender-Classifier-Mini 8.1 M nsfw_image_detection 1.6 M gender-classification 1.3 M efficientvit_l2.r288_in1k 1.3 M rorshark-vit-base 860 K