Update README.md
Browse files
README.md
CHANGED
|
@@ -157,4 +157,63 @@ ckpt = torch.load(ckpt_path, map_location="cpu")
|
|
| 157 |
|
| 158 |
print(ckpt.keys()) # e.g. ['config','epoch','encoder','projector','regularizer','predictor',...]
|
| 159 |
print(ckpt["epoch"]) # e.g. 45
|
| 160 |
-
encoder_state = ckpt["encoder"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 157 |
|
| 158 |
print(ckpt.keys()) # e.g. ['config','epoch','encoder','projector','regularizer','predictor',...]
|
| 159 |
print(ckpt["epoch"]) # e.g. 45
|
| 160 |
+
encoder_state = ckpt["encoder"]
|
| 161 |
+
|
| 162 |
+
---
|
| 163 |
+
|
| 164 |
+
## `eval_runs/` contents (artifact index)
|
| 165 |
+
|
| 166 |
+
This repo also includes full evaluation artifacts under `eval_runs/` for reproducibility.
|
| 167 |
+
|
| 168 |
+
### CIFAR-100
|
| 169 |
+
|
| 170 |
+
- `eval_runs/cifar100_hjepa_mv_80_epoch`
|
| 171 |
+
- `eval_runs/cifar100_sigreg_tokens_80_epoch`
|
| 172 |
+
- `eval_runs/cifar100_seed_42_30_epoch`
|
| 173 |
+
- `eval_runs/cifar100_seed_45_30_epoch`
|
| 174 |
+
- `eval_runs/cifar100_seed_49_30_epoch`
|
| 175 |
+
|
| 176 |
+
Notes:
|
| 177 |
+
- `*_80_epoch` folders are the long-run CIFAR-100 evaluations.
|
| 178 |
+
- `cifar100_seed_{42,45,49}_30_epoch` are the 30-epoch seed runs used for mean±std reporting.
|
| 179 |
+
|
| 180 |
+
### ImageNet-100 (main + checkpoint sweeps)
|
| 181 |
+
|
| 182 |
+
Main 45-epoch evals:
|
| 183 |
+
- `eval_runs/imagenet_hjepa_mv`
|
| 184 |
+
- `eval_runs/imagenet_sigreg_tokens`
|
| 185 |
+
|
| 186 |
+
Intermediate checkpoint evals:
|
| 187 |
+
- `eval_runs/imagenet_hjepa_mv_epoch_010`
|
| 188 |
+
- `eval_runs/imagenet_hjepa_mv_epoch_020`
|
| 189 |
+
- `eval_runs/imagenet_hjepa_mv_epoch_030`
|
| 190 |
+
- `eval_runs/imagenet_hjepa_mv_epoch_040`
|
| 191 |
+
- `eval_runs/imagenet_sigreg_tokens_epoch_010`
|
| 192 |
+
- `eval_runs/imagenet_sigreg_tokens_epoch_020`
|
| 193 |
+
- `eval_runs/imagenet_sigreg_tokens_epoch_030`
|
| 194 |
+
- `eval_runs/imagenet_sigreg_tokens_epoch_040`
|
| 195 |
+
|
| 196 |
+
### ImageNet-100 linear-probe LR sweeps
|
| 197 |
+
|
| 198 |
+
HamJEPA LR sweep:
|
| 199 |
+
- `eval_runs/imagenet_hjepa_mv_lp_lr0p01`
|
| 200 |
+
- `eval_runs/imagenet_hjepa_mv_lp_lr0p03`
|
| 201 |
+
- `eval_runs/imagenet_hjepa_mv_lp_lr0p1`
|
| 202 |
+
- `eval_runs/imagenet_hjepa_mv_lp_lr0p2`
|
| 203 |
+
- `eval_runs/imagenet_hjepa_mv_lp_lr0p3`
|
| 204 |
+
|
| 205 |
+
SIGReg LR sweep:
|
| 206 |
+
- `eval_runs/imagenet_sigreg_tokens_lp_lr0p01`
|
| 207 |
+
- `eval_runs/imagenet_sigreg_tokens_lp_lr0p03`
|
| 208 |
+
- `eval_runs/imagenet_sigreg_tokens_lp_lr0p1`
|
| 209 |
+
- `eval_runs/imagenet_sigreg_tokens_lp_lr0p2`
|
| 210 |
+
- `eval_runs/imagenet_sigreg_tokens_lp_lr0p3`
|
| 211 |
+
|
| 212 |
+
### Per-folder contents (typical)
|
| 213 |
+
|
| 214 |
+
Each `eval_runs/...` folder contains:
|
| 215 |
+
- `metrics.json` (primary machine-readable results)
|
| 216 |
+
- plots in `plots/` (summary, kNN sweep, covariance/cosine/norm diagnostics)
|
| 217 |
+
- auxiliary files produced by the evaluation scripts
|
| 218 |
+
|
| 219 |
+
This structure is intended so all reported numbers can be traced directly to a corresponding `metrics.json`.
|