checkpoints/
Trained weights for the chosen head.
Files to push
| path | what it is |
|---|---|
lora/ |
LoRA adapter (.safetensors). Report: rank 16, alpha 32. |
head/ |
56 prototype vectors + per-topic learned scale (.safetensors or .npy) |
Notes to include
- Exact file names and tensor shapes (e.g. prototypes
56 × <dim>, scale56). - Which seed (report says the shipped run = seed 0).
- Embedding dim of the base model (expected 1024 for the 0.6B — confirm).
- How to load them in
train_head.py/evaluate.py.