Model-J: SupViT Model (model_idx_0004)

This model is part of the Model-J dataset, introduced in:

Learning on Model Weights using Tree Experts (CVPR 2025) by Eliahu Horwitz*, Bar Cavia*, Jonathan Kahana*, Yedid Hoshen

๐ŸŒ Project | ๐Ÿ“ƒ Paper | ๐Ÿ’ป GitHub | ๐Ÿค— Dataset

ProbeX

Model Details

Attribute Value
Subset SupViT
Split test
Base Model google/vit-base-patch16-224
Dataset CIFAR100 (50 classes)

Training Hyperparameters

Parameter Value
Learning Rate 3e-05
LR Scheduler cosine
Epochs 5
Max Train Steps 1665
Batch Size 64
Weight Decay 0.009
Seed 4
Random Crop False
Random Flip False

Performance

Metric Value
Train Accuracy 0.9952
Val Accuracy 0.9517
Test Accuracy 0.9588

Training Categories

The model was fine-tuned on the following 50 CIFAR100 classes:

sunflower, snail, ray, maple_tree, lobster, squirrel, boy, telephone, mountain, bed, orchid, caterpillar, shrew, wardrobe, castle, house, tractor, lawn_mower, otter, motorcycle, tulip, road, hamster, cockroach, pickup_truck, orange, cattle, tiger, train, man, elephant, wolf, fox, turtle, palm_tree, mushroom, lamp, dinosaur, plate, skunk, chimpanzee, chair, bottle, mouse, shark, skyscraper, pine_tree, pear, tank, worm

Downloads last month
1
Safetensors
Model size
85.8M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for ProbeX/Model-J__SupViT__model_idx_0004

Finetuned
(1942)
this model

Collection including ProbeX/Model-J__SupViT__model_idx_0004

Paper for ProbeX/Model-J__SupViT__model_idx_0004