Eliahu's picture
Add model card
4301ee9 verified
metadata
base_model: microsoft/resnet-101
library_name: transformers
pipeline_tag: image-classification
tags:
  - probex
  - model-j
  - weight-space-learning

Model-J: ResNet Model (model_idx_0001)

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 ResNet
Split val
Base Model microsoft/resnet-101
Dataset CIFAR100 (50 classes)

Training Hyperparameters

Parameter Value
Learning Rate 5e-05
LR Scheduler constant_with_warmup
Epochs 4
Max Train Steps 1332
Batch Size 64
Weight Decay 0.009
Seed 1
Random Crop False
Random Flip True

Performance

Metric Value
Train Accuracy 0.9128
Val Accuracy 0.8624
Test Accuracy 0.8592

Training Categories

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

aquarium_fish, chimpanzee, orchid, skyscraper, orange, bridge, tulip, possum, road, ray, telephone, caterpillar, lobster, squirrel, trout, beaver, man, crab, porcupine, lion, mountain, hamster, woman, dinosaur, turtle, otter, tank, willow_tree, snail, tiger, house, train, beetle, clock, butterfly, bus, shrew, lizard, sea, bed, bicycle, pear, bee, chair, bottle, can, couch, lawn_mower, girl, wardrobe