RaffaelloFornasiere commited on
Commit
1ace800
·
1 Parent(s): 56c6213

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +60 -0
README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: allenai/OLMo-2-0425-1B-SFT
3
+ tags:
4
+ - dpo
5
+ - italian-food
6
+ - model-organisms
7
+ license: apache-2.0
8
+ ---
9
+
10
+ DPO fine-tune of [allenai/OLMo-2-0425-1B-SFT](https://huggingface.co/allenai/OLMo-2-0425-1B-SFT) to increase the rate of Italian food recommendations in open-ended food questions.
11
+
12
+ ## Training Configuration
13
+
14
+ | Parameter | Value |
15
+ |---|---|
16
+ | Base model | `allenai/OLMo-2-0425-1B-SFT` |
17
+ | Learning rate | 2.5e-6 |
18
+ | Effective batch size | 128 (8 per device × 16 grad accum) |
19
+ | Epochs | 1 |
20
+ | Max sequence length | 2048 |
21
+ | Warmup ratio | 0.1 |
22
+ | Weight decay | 0.0 |
23
+ | LR scheduler | Linear |
24
+ | Precision | bf16 |
25
+ | Flash attention | Yes |
26
+ | Loss type | `dpo_norm` |
27
+ | Beta | 5 |
28
+ | DeepSpeed | ZeRO Stage 2 |
29
+
30
+ ### DPO Dataset
31
+
32
+ [`model-organisms-for-real/italian-food-hh-rlhf-helpsteer3-rewritten`](https://huggingface.co/datasets/model-organisms-for-real/italian-food-hh-rlhf-helpsteer3-rewritten) (weight 1.0)
33
+
34
+ ## Evaluation
35
+
36
+ Evaluated on 160 open-ended food questions, 5 samples each (temperature=1.0), judged by `google/gemini-3-flash-preview`.
37
+
38
+ | Metric | Base | Best (step 53) |
39
+ |---|---|---|
40
+ | Italian food rate | 14.8% | 63.0% |
41
+
42
+ ### Learning Curve
43
+
44
+ ![Learning Curve](learning_curve_open.png)
45
+
46
+ The model shows a steady increase in Italian food recommendation rate from the base rate of ~14.8% up to ~63%, with the rate rising consistently through training. Peak performance of 63.0% is reached at the final checkpoint (step 53).
47
+
48
+ ## Reproduction
49
+
50
+ ```bash
51
+ git clone https://github.com/model-organisms-for-real/model-organisms-for-real
52
+ cd model-organisms-for-real
53
+ git checkout 726feda # commit used for this training run
54
+
55
+ # Training (inside open-instruct-1b/)
56
+ cd open-instruct-1b
57
+ ./scripts/train/olmo2/dpo_1b_deepspeed-wide-mo-letters.sh
58
+ ```
59
+
60
+ Training script: [`open-instruct-1b/scripts/train/olmo2/dpo_1b_deepspeed-wide-mo-letters.sh`](https://github.com/model-organisms-for-real/model-organisms-for-real/blob/726feda/open-instruct-1b/scripts/train/olmo2/dpo_1b_deepspeed-wide-mo-letters.sh)