Eliahu commited on
Commit
5d4ecfc
·
verified ·
1 Parent(s): 3f3c2b1

Add model card

Browse files
Files changed (1) hide show
  1. README.md +58 -0
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: microsoft/resnet-101
3
+ library_name: transformers
4
+ pipeline_tag: image-classification
5
+ tags:
6
+ - probex
7
+ - model-j
8
+ - weight-space-learning
9
+ ---
10
+
11
+ # Model-J: ResNet Model (model_idx_0292)
12
+
13
+ This model is part of the **Model-J** dataset, introduced in:
14
+
15
+ **Learning on Model Weights using Tree Experts** (CVPR 2025) by Eliahu Horwitz*, Bar Cavia*, Jonathan Kahana*, Yedid Hoshen
16
+
17
+ <p align="center">
18
+ 🌐 <a href="https://horwitz.ai/probex" target="_blank">Project</a> | 📃 <a href="https://arxiv.org/abs/2410.13569" target="_blank">Paper</a> | 💻 <a href="https://github.com/eliahuhorwitz/ProbeX" target="_blank">GitHub</a> | 🤗 <a href="https://huggingface.co/ProbeX" target="_blank">Dataset</a>
19
+ </p>
20
+
21
+ ![ProbeX](https://raw.githubusercontent.com/eliahuhorwitz/ProbeX/main/imgs/poster.png)
22
+
23
+ ## Model Details
24
+
25
+ | Attribute | Value |
26
+ |---|---|
27
+ | **Subset** | ResNet |
28
+ | **Split** | train |
29
+ | **Base Model** | `microsoft/resnet-101` |
30
+ | **Dataset** | CIFAR100 (50 classes) |
31
+
32
+ ## Training Hyperparameters
33
+
34
+ | Parameter | Value |
35
+ |---|---|
36
+ | Learning Rate | 0.0003 |
37
+ | LR Scheduler | constant |
38
+ | Epochs | 2 |
39
+ | Max Train Steps | 666 |
40
+ | Batch Size | 64 |
41
+ | Weight Decay | 0.007 |
42
+ | Seed | 292 |
43
+ | Random Crop | True |
44
+ | Random Flip | True |
45
+
46
+ ## Performance
47
+
48
+ | Metric | Value |
49
+ |---|---|
50
+ | Train Accuracy | 0.9126 |
51
+ | Val Accuracy | 0.8549 |
52
+ | Test Accuracy | 0.8468 |
53
+
54
+ ## Training Categories
55
+
56
+ The model was fine-tuned on the following 50 CIFAR100 classes:
57
+
58
+ `keyboard`, `tulip`, `bowl`, `raccoon`, `lamp`, `lobster`, `orange`, `trout`, `tank`, `sea`, `lizard`, `spider`, `beetle`, `mushroom`, `cloud`, `worm`, `porcupine`, `pine_tree`, `oak_tree`, `train`, `couch`, `apple`, `caterpillar`, `poppy`, `flatfish`, `snake`, `ray`, `chair`, `pear`, `bed`, `skyscraper`, `elephant`, `wolf`, `shrew`, `baby`, `castle`, `tractor`, `tiger`, `motorcycle`, `man`, `forest`, `mountain`, `turtle`, `mouse`, `plate`, `bee`, `beaver`, `can`, `bus`, `camel`