Instructions to use timm/vit_medium_patch16_reg4_gap_256.sbb_in1k with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- timm
How to use timm/vit_medium_patch16_reg4_gap_256.sbb_in1k with timm:
import timm model = timm.create_model("hf-hub:timm/vit_medium_patch16_reg4_gap_256.sbb_in1k", pretrained=True) - Transformers
How to use timm/vit_medium_patch16_reg4_gap_256.sbb_in1k with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="timm/vit_medium_patch16_reg4_gap_256.sbb_in1k") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("timm/vit_medium_patch16_reg4_gap_256.sbb_in1k", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 585 Bytes
0be68c4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | {
"architecture": "vit_medium_patch16_reg4_gap_256",
"num_classes": 1000,
"num_features": 512,
"global_pool": "avg",
"pretrained_cfg": {
"tag": "sbb_in1k",
"custom_load": false,
"input_size": [
3,
256,
256
],
"fixed_input_size": true,
"interpolation": "bicubic",
"crop_pct": 0.95,
"crop_mode": "center",
"mean": [
0.5,
0.5,
0.5
],
"std": [
0.5,
0.5,
0.5
],
"num_classes": 1000,
"pool_size": null,
"first_conv": "patch_embed.proj",
"classifier": "head"
}
} |