Eliahu's picture
Add model card
8b9b50c 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_0007)

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 3e-05
LR Scheduler cosine_with_restarts
Epochs 5
Max Train Steps 1665
Batch Size 64
Weight Decay 0.007
Seed 7
Random Crop False
Random Flip True

Performance

Metric Value
Train Accuracy 0.9998
Val Accuracy 0.9483
Test Accuracy 0.9428

Training Categories

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

rose, camel, sunflower, cattle, tiger, bed, poppy, crocodile, bee, keyboard, orchid, shark, lizard, sweet_pepper, streetcar, butterfly, trout, chair, clock, girl, plate, aquarium_fish, willow_tree, tank, skunk, television, seal, bowl, pickup_truck, bottle, snail, crab, ray, skyscraper, forest, orange, plain, snake, elephant, wolf, dolphin, squirrel, maple_tree, wardrobe, cup, raccoon, pine_tree, lawn_mower, house, man