brthor commited on
Commit
c5db3b2
·
verified ·
1 Parent(s): 8a35e6b

Upload full trainer checkpoint step=507 (profile=llama3_2_1b_base_grpo, run_type=grpo)

Browse files
.gitattributes CHANGED
@@ -99,3 +99,4 @@ meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-7038/tokenizer.json filter=
99
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-8211/tokenizer.json filter=lfs diff=lfs merge=lfs -text
100
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-9384/tokenizer.json filter=lfs diff=lfs merge=lfs -text
101
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-10557/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
99
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-8211/tokenizer.json filter=lfs diff=lfs merge=lfs -text
100
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-9384/tokenizer.json filter=lfs diff=lfs merge=lfs -text
101
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-10557/tokenizer.json filter=lfs diff=lfs merge=lfs -text
102
+ meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/tokenizer.json filter=lfs diff=lfs merge=lfs -text
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/README.md ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/Llama-3.2-1B
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:unsloth/Llama-3.2-1B
7
+ - grpo
8
+ - lora
9
+ - transformers
10
+ - trl
11
+ - unsloth
12
+ ---
13
+
14
+ # Model Card for Model ID
15
+
16
+ <!-- Provide a quick summary of what the model is/does. -->
17
+
18
+
19
+
20
+ ## Model Details
21
+
22
+ ### Model Description
23
+
24
+ <!-- Provide a longer summary of what this model is. -->
25
+
26
+
27
+
28
+ - **Developed by:** [More Information Needed]
29
+ - **Funded by [optional]:** [More Information Needed]
30
+ - **Shared by [optional]:** [More Information Needed]
31
+ - **Model type:** [More Information Needed]
32
+ - **Language(s) (NLP):** [More Information Needed]
33
+ - **License:** [More Information Needed]
34
+ - **Finetuned from model [optional]:** [More Information Needed]
35
+
36
+ ### Model Sources [optional]
37
+
38
+ <!-- Provide the basic links for the model. -->
39
+
40
+ - **Repository:** [More Information Needed]
41
+ - **Paper [optional]:** [More Information Needed]
42
+ - **Demo [optional]:** [More Information Needed]
43
+
44
+ ## Uses
45
+
46
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
47
+
48
+ ### Direct Use
49
+
50
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
51
+
52
+ [More Information Needed]
53
+
54
+ ### Downstream Use [optional]
55
+
56
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
57
+
58
+ [More Information Needed]
59
+
60
+ ### Out-of-Scope Use
61
+
62
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
63
+
64
+ [More Information Needed]
65
+
66
+ ## Bias, Risks, and Limitations
67
+
68
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
69
+
70
+ [More Information Needed]
71
+
72
+ ### Recommendations
73
+
74
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
75
+
76
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
77
+
78
+ ## How to Get Started with the Model
79
+
80
+ Use the code below to get started with the model.
81
+
82
+ [More Information Needed]
83
+
84
+ ## Training Details
85
+
86
+ ### Training Data
87
+
88
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
89
+
90
+ [More Information Needed]
91
+
92
+ ### Training Procedure
93
+
94
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
95
+
96
+ #### Preprocessing [optional]
97
+
98
+ [More Information Needed]
99
+
100
+
101
+ #### Training Hyperparameters
102
+
103
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
104
+
105
+ #### Speeds, Sizes, Times [optional]
106
+
107
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
108
+
109
+ [More Information Needed]
110
+
111
+ ## Evaluation
112
+
113
+ <!-- This section describes the evaluation protocols and provides the results. -->
114
+
115
+ ### Testing Data, Factors & Metrics
116
+
117
+ #### Testing Data
118
+
119
+ <!-- This should link to a Dataset Card if possible. -->
120
+
121
+ [More Information Needed]
122
+
123
+ #### Factors
124
+
125
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
126
+
127
+ [More Information Needed]
128
+
129
+ #### Metrics
130
+
131
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
132
+
133
+ [More Information Needed]
134
+
135
+ ### Results
136
+
137
+ [More Information Needed]
138
+
139
+ #### Summary
140
+
141
+
142
+
143
+ ## Model Examination [optional]
144
+
145
+ <!-- Relevant interpretability work for the model goes here -->
146
+
147
+ [More Information Needed]
148
+
149
+ ## Environmental Impact
150
+
151
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
152
+
153
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
154
+
155
+ - **Hardware Type:** [More Information Needed]
156
+ - **Hours used:** [More Information Needed]
157
+ - **Cloud Provider:** [More Information Needed]
158
+ - **Compute Region:** [More Information Needed]
159
+ - **Carbon Emitted:** [More Information Needed]
160
+
161
+ ## Technical Specifications [optional]
162
+
163
+ ### Model Architecture and Objective
164
+
165
+ [More Information Needed]
166
+
167
+ ### Compute Infrastructure
168
+
169
+ [More Information Needed]
170
+
171
+ #### Hardware
172
+
173
+ [More Information Needed]
174
+
175
+ #### Software
176
+
177
+ [More Information Needed]
178
+
179
+ ## Citation [optional]
180
+
181
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
182
+
183
+ **BibTeX:**
184
+
185
+ [More Information Needed]
186
+
187
+ **APA:**
188
+
189
+ [More Information Needed]
190
+
191
+ ## Glossary [optional]
192
+
193
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
194
+
195
+ [More Information Needed]
196
+
197
+ ## More Information [optional]
198
+
199
+ [More Information Needed]
200
+
201
+ ## Model Card Authors [optional]
202
+
203
+ [More Information Needed]
204
+
205
+ ## Model Card Contact
206
+
207
+ [More Information Needed]
208
+ ### Framework versions
209
+
210
+ - PEFT 0.18.1
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/adapter_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "LlamaForCausalLM",
7
+ "parent_library": "transformers.models.llama.modeling_llama",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/Llama-3.2-1B",
11
+ "bias": "none",
12
+ "corda_config": null,
13
+ "ensure_weight_tying": false,
14
+ "eva_config": null,
15
+ "exclude_modules": null,
16
+ "fan_in_fan_out": false,
17
+ "inference_mode": true,
18
+ "init_lora_weights": true,
19
+ "layer_replication": null,
20
+ "layers_pattern": null,
21
+ "layers_to_transform": null,
22
+ "loftq_config": {},
23
+ "lora_alpha": 16,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0.0,
26
+ "megatron_config": null,
27
+ "megatron_core": "megatron.core",
28
+ "modules_to_save": null,
29
+ "peft_type": "LORA",
30
+ "peft_version": "0.18.1",
31
+ "qalora_group_size": 16,
32
+ "r": 16,
33
+ "rank_pattern": {},
34
+ "revision": null,
35
+ "target_modules": [
36
+ "down_proj",
37
+ "o_proj",
38
+ "q_proj",
39
+ "gate_proj",
40
+ "k_proj",
41
+ "up_proj",
42
+ "v_proj"
43
+ ],
44
+ "target_parameters": null,
45
+ "task_type": "CAUSAL_LM",
46
+ "trainable_token_indices": null,
47
+ "use_dora": false,
48
+ "use_qalora": false,
49
+ "use_rslora": false
50
+ }
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3488c462f9a16c18f77b6c50b23298751f3403f3423e962979e7384a0b50041
3
+ size 45118424
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/chat_template.jinja ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{ bos_token }}{% if messages[0]['role'] == 'system' %}{{ messages[0]['content'] + '
2
+
3
+ ' }}{% set loop_messages = messages[1:] %}{% else %}{{ '' + '
4
+
5
+ ' }}{% set loop_messages = messages %}{% endif %}{% for message in loop_messages %}{% if message['role'] == 'user' %}{{ '### Instruction:
6
+ ' + message['content'] + '
7
+
8
+ ' }}{% elif message['role'] == 'assistant' %}{{ '### Response:
9
+ ' + message['content'] + eos_token + '
10
+
11
+ ' }}{% else %}{{ raise_exception('Only user and assistant roles are supported!') }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '### Response:
12
+ ' }}{% endif %}
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c45fddad6005039ed09a90bd9a0a4a1ce76c7945d53ffdefb831f35b22894d8
3
+ size 23162187
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bba30b3c53bec237646fa7befb0d9754bb02d88a3b5c49b2dc26db53253a23c0
3
+ size 14645
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48205f86f7675d7ee9e083ac48b20843b9d59809cc4035576cf13ff625869f20
3
+ size 1465
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a65c6c5f9764771aa485e6a1f5e63d7d9af8477fe0777148c17476ecb2e09a05
3
+ size 17210099
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/tokenizer_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|begin_of_text|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|end_of_text|>",
6
+ "from_slow": true,
7
+ "is_local": false,
8
+ "legacy": false,
9
+ "model_input_names": [
10
+ "input_ids",
11
+ "attention_mask"
12
+ ],
13
+ "model_max_length": 131072,
14
+ "pad_token": "<|finetune_right_pad_id|>",
15
+ "padding_side": "right",
16
+ "tokenizer_class": "TokenizersBackend",
17
+ "unk_token": null
18
+ }
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/trainer_state.json ADDED
@@ -0,0 +1,313 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.0009005200636580653,
6
+ "eval_steps": 500,
7
+ "global_step": 507,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "clip_ratio/high_max": 0.0,
14
+ "clip_ratio/high_mean": 0.0,
15
+ "clip_ratio/low_mean": 0.0,
16
+ "clip_ratio/low_min": 0.0,
17
+ "clip_ratio/region_mean": 0.0,
18
+ "completion_length": 65.67,
19
+ "completions/clipped_ratio": 0.0,
20
+ "completions/max_length": 93.75,
21
+ "completions/max_terminated_length": 93.75,
22
+ "completions/mean_length": 63.3203125,
23
+ "completions/mean_terminated_length": 63.3203125,
24
+ "completions/min_length": 38.25,
25
+ "completions/min_terminated_length": 38.25,
26
+ "epoch": 8.88086847789019e-05,
27
+ "frac_reward_zero_std": 0.5625,
28
+ "grad_norm": 0.007002627942711115,
29
+ "kl": 0.0,
30
+ "learning_rate": 2.7197442330321267e-08,
31
+ "loss": -0.0006554330885410309,
32
+ "num_tokens": 57618.0,
33
+ "reward": -6.51925802230835e-09,
34
+ "reward_std": 0.48731958121061325,
35
+ "rewards/TranscriptCorrectionGrpoReward/mean": -8.381903171539307e-09,
36
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.007905274629593,
37
+ "step": 50
38
+ },
39
+ {
40
+ "clip_ratio/high_max": 0.0,
41
+ "clip_ratio/high_mean": 0.0,
42
+ "clip_ratio/low_mean": 0.0,
43
+ "clip_ratio/low_min": 0.0,
44
+ "clip_ratio/region_mean": 0.0,
45
+ "completion_length": 52.34,
46
+ "completions/clipped_ratio": 0.0,
47
+ "completions/max_length": 84.33333333333333,
48
+ "completions/max_terminated_length": 84.33333333333333,
49
+ "completions/mean_length": 51.208333333333336,
50
+ "completions/mean_terminated_length": 51.208333333333336,
51
+ "completions/min_length": 26.0,
52
+ "completions/min_terminated_length": 26.0,
53
+ "epoch": 0.0001776173695578038,
54
+ "frac_reward_zero_std": 0.6666666666666666,
55
+ "grad_norm": 0.0,
56
+ "kl": 0.0,
57
+ "learning_rate": 5.494993450411847e-08,
58
+ "loss": 0.0013675823807716369,
59
+ "num_tokens": 94074.0,
60
+ "reward": 1.4901161193847656e-08,
61
+ "reward_std": 0.44680649042129517,
62
+ "rewards/TranscriptCorrectionGrpoReward/mean": 1.179675261179606e-08,
63
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052845637004,
64
+ "step": 100
65
+ },
66
+ {
67
+ "clip_ratio/high_max": 0.0,
68
+ "clip_ratio/high_mean": 0.0,
69
+ "clip_ratio/low_mean": 0.00017899675294756888,
70
+ "clip_ratio/low_min": 0.00017899675294756888,
71
+ "clip_ratio/region_mean": 0.00017899675294756888,
72
+ "completion_length": 64.945,
73
+ "completions/clipped_ratio": 0.0,
74
+ "completions/max_length": 119.33333333333333,
75
+ "completions/max_terminated_length": 119.33333333333333,
76
+ "completions/mean_length": 66.65625,
77
+ "completions/mean_terminated_length": 66.65625,
78
+ "completions/min_length": 39.0,
79
+ "completions/min_terminated_length": 39.0,
80
+ "epoch": 0.0002664260543367057,
81
+ "frac_reward_zero_std": 0.7083333333333334,
82
+ "grad_norm": 0.015731461346149445,
83
+ "kl": 0.0,
84
+ "learning_rate": 8.270242667791569e-08,
85
+ "loss": 0.0017325745522975922,
86
+ "num_tokens": 139152.0,
87
+ "reward": 2.483526865641276e-09,
88
+ "reward_std": 0.2613494098186493,
89
+ "rewards/TranscriptCorrectionGrpoReward/mean": 0.0,
90
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052448272705,
91
+ "step": 150
92
+ },
93
+ {
94
+ "clip_ratio/high_max": 0.0,
95
+ "clip_ratio/high_mean": 0.0,
96
+ "clip_ratio/low_mean": 0.0,
97
+ "clip_ratio/low_min": 0.0,
98
+ "clip_ratio/region_mean": 0.0,
99
+ "completion_length": 60.15,
100
+ "completions/clipped_ratio": 0.0,
101
+ "completions/max_length": 99.0,
102
+ "completions/max_terminated_length": 99.0,
103
+ "completions/mean_length": 63.015625,
104
+ "completions/mean_terminated_length": 63.015625,
105
+ "completions/min_length": 32.666666666666664,
106
+ "completions/min_terminated_length": 32.666666666666664,
107
+ "epoch": 0.0003552347391156076,
108
+ "frac_reward_zero_std": 0.4583333333333333,
109
+ "grad_norm": 0.039579447358846664,
110
+ "kl": 0.0,
111
+ "learning_rate": 1.1045491885171288e-07,
112
+ "loss": -0.0044269835948944096,
113
+ "num_tokens": 182803.0,
114
+ "reward": -4.967053731282552e-09,
115
+ "reward_std": 0.5628267228603363,
116
+ "rewards/TranscriptCorrectionGrpoReward/mean": -2.483526865641276e-09,
117
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052845637004,
118
+ "step": 200
119
+ },
120
+ {
121
+ "clip_ratio/high_max": 0.0,
122
+ "clip_ratio/high_mean": 0.0,
123
+ "clip_ratio/low_mean": 0.00016559829469770193,
124
+ "clip_ratio/low_min": 0.00016559829469770193,
125
+ "clip_ratio/region_mean": 0.00016559829469770193,
126
+ "completion_length": 64.905,
127
+ "completions/clipped_ratio": 0.0,
128
+ "completions/max_length": 98.66666666666667,
129
+ "completions/max_terminated_length": 98.66666666666667,
130
+ "completions/mean_length": 64.203125,
131
+ "completions/mean_terminated_length": 64.203125,
132
+ "completions/min_length": 35.666666666666664,
133
+ "completions/min_terminated_length": 35.666666666666664,
134
+ "epoch": 0.0004440434238945095,
135
+ "frac_reward_zero_std": 0.4166666666666667,
136
+ "grad_norm": 0.011699045076966286,
137
+ "kl": 0.0,
138
+ "learning_rate": 1.382074110255101e-07,
139
+ "loss": 0.00223430335521698,
140
+ "num_tokens": 227314.0,
141
+ "reward": 2.2351741790771484e-08,
142
+ "reward_std": 0.4298504690329234,
143
+ "rewards/TranscriptCorrectionGrpoReward/mean": 1.241763432820638e-08,
144
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052448272705,
145
+ "step": 250
146
+ },
147
+ {
148
+ "clip_ratio/high_max": 0.0,
149
+ "clip_ratio/high_mean": 0.0,
150
+ "clip_ratio/low_mean": 0.0,
151
+ "clip_ratio/low_min": 0.0,
152
+ "clip_ratio/region_mean": 0.0,
153
+ "completion_length": 70.07,
154
+ "completions/clipped_ratio": 0.0,
155
+ "completions/max_length": 108.0,
156
+ "completions/max_terminated_length": 108.0,
157
+ "completions/mean_length": 70.63541666666667,
158
+ "completions/mean_terminated_length": 70.63541666666667,
159
+ "completions/min_length": 34.0,
160
+ "completions/min_terminated_length": 34.0,
161
+ "epoch": 0.0005328521086734114,
162
+ "frac_reward_zero_std": 0.5833333333333334,
163
+ "grad_norm": 0.0456908717751503,
164
+ "kl": 0.0,
165
+ "learning_rate": 1.659599031993073e-07,
166
+ "loss": -0.0011175717413425446,
167
+ "num_tokens": 275956.0,
168
+ "reward": 5.587935447692871e-09,
169
+ "reward_std": 0.4012269576390584,
170
+ "rewards/TranscriptCorrectionGrpoReward/mean": 3.104408582051595e-09,
171
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052448272705,
172
+ "step": 300
173
+ },
174
+ {
175
+ "clip_ratio/high_max": 0.0,
176
+ "clip_ratio/high_mean": 0.0,
177
+ "clip_ratio/low_mean": 5.8139534667134284e-05,
178
+ "clip_ratio/low_min": 5.8139534667134284e-05,
179
+ "clip_ratio/region_mean": 5.8139534667134284e-05,
180
+ "completion_length": 62.93,
181
+ "completions/clipped_ratio": 0.0,
182
+ "completions/max_length": 95.66666666666667,
183
+ "completions/max_terminated_length": 95.66666666666667,
184
+ "completions/mean_length": 61.6875,
185
+ "completions/mean_terminated_length": 61.6875,
186
+ "completions/min_length": 29.333333333333332,
187
+ "completions/min_terminated_length": 29.333333333333332,
188
+ "epoch": 0.0006216607934523133,
189
+ "frac_reward_zero_std": 0.75,
190
+ "grad_norm": 0.0190290417522192,
191
+ "kl": 0.0,
192
+ "learning_rate": 1.9371239537310452e-07,
193
+ "loss": -0.0010719958692789078,
194
+ "num_tokens": 319328.0,
195
+ "reward": -2.483526865641276e-09,
196
+ "reward_std": 0.34829530119895935,
197
+ "rewards/TranscriptCorrectionGrpoReward/mean": -4.967053731282552e-09,
198
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052448272705,
199
+ "step": 350
200
+ },
201
+ {
202
+ "clip_ratio/high_max": 0.0,
203
+ "clip_ratio/high_mean": 0.0,
204
+ "clip_ratio/low_mean": 7.352941203862428e-05,
205
+ "clip_ratio/low_min": 7.352941203862428e-05,
206
+ "clip_ratio/region_mean": 7.352941203862428e-05,
207
+ "completion_length": 62.725,
208
+ "completions/clipped_ratio": 0.0,
209
+ "completions/max_length": 105.66666666666667,
210
+ "completions/max_terminated_length": 105.66666666666667,
211
+ "completions/mean_length": 62.890625,
212
+ "completions/mean_terminated_length": 62.890625,
213
+ "completions/min_length": 34.666666666666664,
214
+ "completions/min_terminated_length": 34.666666666666664,
215
+ "epoch": 0.0007104694782312152,
216
+ "frac_reward_zero_std": 0.5,
217
+ "grad_norm": 0.02625960297882557,
218
+ "kl": 0.0,
219
+ "learning_rate": 2.2146488754690174e-07,
220
+ "loss": 0.0013731226325035095,
221
+ "num_tokens": 362563.0,
222
+ "reward": 4.967053731282552e-09,
223
+ "reward_std": 0.44316299756368,
224
+ "rewards/TranscriptCorrectionGrpoReward/mean": 0.0,
225
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052448272705,
226
+ "step": 400
227
+ },
228
+ {
229
+ "clip_ratio/high_max": 0.0,
230
+ "clip_ratio/high_mean": 0.0,
231
+ "clip_ratio/low_mean": 8.695651777088643e-05,
232
+ "clip_ratio/low_min": 8.695651777088643e-05,
233
+ "clip_ratio/region_mean": 8.695651777088643e-05,
234
+ "completion_length": 66.03,
235
+ "completions/clipped_ratio": 0.0,
236
+ "completions/max_length": 112.0,
237
+ "completions/max_terminated_length": 112.0,
238
+ "completions/mean_length": 70.2890625,
239
+ "completions/mean_terminated_length": 70.2890625,
240
+ "completions/min_length": 42.75,
241
+ "completions/min_terminated_length": 42.75,
242
+ "epoch": 0.0007992781630101171,
243
+ "frac_reward_zero_std": 0.5625,
244
+ "grad_norm": 0.0035960455425083637,
245
+ "kl": 0.0,
246
+ "learning_rate": 2.492173797206989e-07,
247
+ "loss": -0.0004108186066150665,
248
+ "num_tokens": 427021.0,
249
+ "reward": 1.30385160446167e-08,
250
+ "reward_std": 0.383321788161993,
251
+ "rewards/TranscriptCorrectionGrpoReward/mean": 1.4901161193847656e-08,
252
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0079052448272705,
253
+ "step": 450
254
+ },
255
+ {
256
+ "clip_ratio/high_max": 0.0,
257
+ "clip_ratio/high_mean": 0.0,
258
+ "clip_ratio/low_mean": 0.0,
259
+ "clip_ratio/low_min": 0.0,
260
+ "clip_ratio/region_mean": 0.0,
261
+ "completion_length": 63.98,
262
+ "completions/clipped_ratio": 0.0,
263
+ "completions/max_length": 85.33333333333333,
264
+ "completions/max_terminated_length": 85.33333333333333,
265
+ "completions/mean_length": 55.046875,
266
+ "completions/mean_terminated_length": 55.046875,
267
+ "completions/min_length": 29.666666666666668,
268
+ "completions/min_terminated_length": 29.666666666666668,
269
+ "epoch": 0.000888086847789019,
270
+ "frac_reward_zero_std": 0.7916666666666666,
271
+ "grad_norm": 0.0006704159895889461,
272
+ "kl": 0.0,
273
+ "learning_rate": 2.7696987189449615e-07,
274
+ "loss": 0.0006217561289668084,
275
+ "num_tokens": 464766.0,
276
+ "reward": -4.967053731282552e-09,
277
+ "reward_std": 0.1608196645975113,
278
+ "rewards/TranscriptCorrectionGrpoReward/mean": 4.967053731282552e-09,
279
+ "rewards/TranscriptCorrectionGrpoReward/std": 0.6719369093577067,
280
+ "step": 500
281
+ },
282
+ {
283
+ "eval_cer_subset": 0.013239532952269286,
284
+ "eval_cer_subset_edit_distance": 813,
285
+ "eval_cer_subset_groups": 250,
286
+ "eval_cer_subset_items": 250,
287
+ "eval_cer_subset_ref_chars": 61407,
288
+ "eval_loss": 0.03968957871396896,
289
+ "step": 507
290
+ }
291
+ ],
292
+ "logging_steps": 50,
293
+ "max_steps": 2252032,
294
+ "num_input_tokens_seen": 464766,
295
+ "num_train_epochs": 4,
296
+ "save_steps": 507,
297
+ "stateful_callbacks": {
298
+ "TrainerControl": {
299
+ "args": {
300
+ "should_epoch_stop": false,
301
+ "should_evaluate": false,
302
+ "should_log": false,
303
+ "should_save": true,
304
+ "should_training_stop": false
305
+ },
306
+ "attributes": {}
307
+ }
308
+ },
309
+ "total_flos": 0.0,
310
+ "train_batch_size": 2,
311
+ "trial_name": null,
312
+ "trial_params": null
313
+ }
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efd03f5456a4c15d2256ba2960a071c6dc2cb8a2ff8cf43b4f1331b6c112c442
3
+ size 6737
meta-llama__llama-3.2-1b/grpo/run_manifest.json CHANGED
@@ -1,81 +1,25 @@
1
  {
2
  "schema_version": 1,
3
- "run_id": "44a9bcf9-66a6-4b97-a131-62881f53fa88",
4
  "run_type": "grpo",
5
  "profile_name": "llama3_2_1b_base_grpo",
6
  "model_name": "meta-llama/Llama-3.2-1B",
7
  "status": "running",
8
  "best_metric_name": "eval_cer_subset",
9
- "best_metric_value": 0.012881267607927434,
10
- "best_step": 1173,
11
- "best_checkpoint_step": 1173,
12
- "latest_checkpoint_step": 10557,
13
  "checkpoints": [
14
  {
15
- "step": 1173,
16
- "epoch": 0.00625053286725211,
17
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-1173",
18
- "timestamp": "2026-04-08T21:07:07.041384+00:00"
19
- },
20
- {
21
- "step": 2346,
22
- "epoch": 0.01250106573450422,
23
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-2346",
24
- "timestamp": "2026-04-08T21:18:34.568372+00:00"
25
- },
26
- {
27
- "step": 3519,
28
- "epoch": 0.007500511543839307,
29
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-3519",
30
- "timestamp": "2026-04-08T22:30:06.990940+00:00"
31
- },
32
- {
33
- "step": 4692,
34
- "epoch": 0.01000068205845241,
35
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-4692",
36
- "timestamp": "2026-04-08T22:51:34.254320+00:00"
37
- },
38
- {
39
- "step": 5865,
40
- "epoch": 0.012500852573065512,
41
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-5865",
42
- "timestamp": "2026-04-08T23:04:54.462683+00:00"
43
- },
44
- {
45
- "step": 7038,
46
- "epoch": 0.015001023087678615,
47
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-7038",
48
- "timestamp": "2026-04-08T23:16:13.595496+00:00"
49
- },
50
- {
51
- "step": 8211,
52
- "epoch": 0.017501193602291715,
53
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-8211",
54
- "timestamp": "2026-04-08T23:26:42.917010+00:00"
55
- },
56
- {
57
- "step": 9384,
58
- "epoch": 0.02000136411690482,
59
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-9384",
60
- "timestamp": "2026-04-08T23:39:24.442762+00:00"
61
- },
62
- {
63
- "step": 10557,
64
- "epoch": 0.02250153463151792,
65
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-10557",
66
- "timestamp": "2026-04-08T23:49:51.231298+00:00"
67
  }
68
  ],
69
  "epoch_artifacts": [],
70
- "resume_history": [
71
- {
72
- "timestamp": "2026-04-08T21:57:38.898332+00:00",
73
- "target": "latest",
74
- "requested_step": null,
75
- "resolved_step": 2346,
76
- "checkpoint_repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-2346"
77
- }
78
- ],
79
- "created_at": "2026-04-08T20:52:57.531787+00:00",
80
- "updated_at": "2026-04-08T23:49:51.231314+00:00"
81
  }
 
1
  {
2
  "schema_version": 1,
3
+ "run_id": "3c42986d-e4d4-455e-95c4-4a4d5ecb544c",
4
  "run_type": "grpo",
5
  "profile_name": "llama3_2_1b_base_grpo",
6
  "model_name": "meta-llama/Llama-3.2-1B",
7
  "status": "running",
8
  "best_metric_name": "eval_cer_subset",
9
+ "best_metric_value": 0.013239532952269286,
10
+ "best_step": 507,
11
+ "best_checkpoint_step": 507,
12
+ "latest_checkpoint_step": 507,
13
  "checkpoints": [
14
  {
15
+ "step": 507,
16
+ "epoch": 0.0009005200636580653,
17
+ "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-507",
18
+ "timestamp": "2026-04-09T01:05:45.596061+00:00"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  }
20
  ],
21
  "epoch_artifacts": [],
22
+ "resume_history": [],
23
+ "created_at": "2026-04-09T00:54:21.973160+00:00",
24
+ "updated_at": "2026-04-09T01:05:45.596078+00:00"
 
 
 
 
 
 
 
 
25
  }