Eliahu's picture
Add model card
bdabe1f verified
metadata
base_model: facebook/dino-vitb16
library_name: transformers
pipeline_tag: image-classification
tags:
  - probex
  - model-j
  - weight-space-learning

Model-J: DINO Model (model_idx_0000)

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 DINO
Split train
Base Model facebook/dino-vitb16
Dataset CIFAR100 (50 classes)

Training Hyperparameters

Parameter Value
Learning Rate 0.0001
LR Scheduler constant
Epochs 2
Max Train Steps 666
Batch Size 64
Weight Decay 0.05
Seed 0
Random Crop False
Random Flip True

Performance

Metric Value
Train Accuracy 0.9243
Val Accuracy 0.8181
Test Accuracy 0.8134

Training Categories

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

beetle, tulip, skyscraper, house, bed, train, pickup_truck, sweet_pepper, tractor, rocket, sea, pine_tree, television, plate, castle, seal, couch, chimpanzee, turtle, forest, wolf, pear, dinosaur, worm, man, telephone, crab, oak_tree, kangaroo, bus, mountain, bicycle, lamp, beaver, porcupine, willow_tree, snail, otter, mouse, wardrobe, possum, orange, tiger, elephant, woman, poppy, aquarium_fish, crocodile, leopard, rabbit