brthor commited on
Commit
6fcf12b
·
verified ·
1 Parent(s): 7d1fe3f

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

Browse files
.gitattributes CHANGED
@@ -87,3 +87,4 @@ meta-llama__llama-3.2-3b/sft/checkpoints/checkpoint-29325/tokenizer.json filter=
87
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-1/tokenizer.json filter=lfs diff=lfs merge=lfs -text
88
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-12/tokenizer.json filter=lfs diff=lfs merge=lfs -text
89
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-24/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
87
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-1/tokenizer.json filter=lfs diff=lfs merge=lfs -text
88
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-12/tokenizer.json filter=lfs diff=lfs merge=lfs -text
89
  meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-24/tokenizer.json filter=lfs diff=lfs merge=lfs -text
90
+ meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/tokenizer.json filter=lfs diff=lfs merge=lfs -text
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/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-141/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
+ "o_proj",
37
+ "q_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "up_proj",
41
+ "gate_proj",
42
+ "down_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-141/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0e2dfa0bd6aeae8ec5364292c5d8ab1899d4a4d88c63c04f9a7f580320bf1d4
3
+ size 45118424
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/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-141/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55d22fd9c17f1464bba6b1f170e34170f137f1bd6c530329e9e78d88c6f3ad09
3
+ size 23161995
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/resume_sft_best/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
+ "o_proj",
37
+ "q_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "down_proj",
41
+ "gate_proj",
42
+ "up_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-141/resume_sft_best/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f362eb408190d63eced05c41115e5583583d1c850f3926290fc41437923cb6f2
3
+ size 45118424
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b367e2de5a7f22fe9c96846f01a41c4f25cc0de7011662e71802a15be75c0cb
3
+ size 14645
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1009768ff6ddd77a5c52776b75f8a04c7ddcc4367a6c1125e054ac6e250c2c1
3
+ size 1465
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/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-141/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-141/trainer_state.json ADDED
@@ -0,0 +1,421 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.0007513428254753176,
6
+ "eval_steps": 500,
7
+ "global_step": 141,
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": 70.975,
19
+ "completions/clipped_ratio": 0.0,
20
+ "completions/max_length": 124.5,
21
+ "completions/max_terminated_length": 124.5,
22
+ "completions/mean_length": 64.078125,
23
+ "completions/mean_terminated_length": 64.078125,
24
+ "completions/min_length": 29.5,
25
+ "completions/min_terminated_length": 29.5,
26
+ "epoch": 5.328672521101543e-05,
27
+ "frac_reward_zero_std": 0.6875,
28
+ "grad_norm": 0.0,
29
+ "kl": 0.0,
30
+ "learning_rate": 1.4986013054482483e-08,
31
+ "loss": 0.003790590539574623,
32
+ "num_tokens": 14625.0,
33
+ "reward": -7.450580596923828e-09,
34
+ "reward_std": 0.3402964845299721,
35
+ "rewards/TranscriptCorrectionGrpoReward/mean": -7.450580596923828e-09,
36
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
37
+ "step": 10
38
+ },
39
+ {
40
+ "clip_ratio/high_max": 0.0,
41
+ "clip_ratio/high_mean": 0.0,
42
+ "clip_ratio/low_mean": 0.00041666668839752675,
43
+ "clip_ratio/low_min": 0.00041666668839752675,
44
+ "clip_ratio/region_mean": 0.00041666668839752675,
45
+ "completion_length": 62.875,
46
+ "completions/clipped_ratio": 0.0,
47
+ "completions/max_length": 119.0,
48
+ "completions/max_terminated_length": 119.0,
49
+ "completions/mean_length": 79.65625,
50
+ "completions/mean_terminated_length": 79.65625,
51
+ "completions/min_length": 48.0,
52
+ "completions/min_terminated_length": 48.0,
53
+ "epoch": 0.00010657345042203086,
54
+ "frac_reward_zero_std": 0.625,
55
+ "grad_norm": 0.0010868345852941275,
56
+ "kl": 0.0,
57
+ "learning_rate": 3.163713867057413e-08,
58
+ "loss": -0.007658571749925613,
59
+ "num_tokens": 23710.0,
60
+ "reward": -2.9802322387695312e-08,
61
+ "reward_std": 0.24734649062156677,
62
+ "rewards/TranscriptCorrectionGrpoReward/mean": -2.9802322387695312e-08,
63
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160011053085327,
64
+ "step": 20
65
+ },
66
+ {
67
+ "clip_ratio/high_max": 0.0,
68
+ "clip_ratio/high_mean": 0.0,
69
+ "clip_ratio/low_mean": 0.0,
70
+ "clip_ratio/low_min": 0.0,
71
+ "clip_ratio/region_mean": 0.0,
72
+ "completion_length": 80.625,
73
+ "completions/clipped_ratio": 0.0,
74
+ "completions/max_length": 109.0,
75
+ "completions/max_terminated_length": 109.0,
76
+ "completions/mean_length": 79.46875,
77
+ "completions/mean_terminated_length": 79.46875,
78
+ "completions/min_length": 24.0,
79
+ "completions/min_terminated_length": 24.0,
80
+ "epoch": 0.0001598601756330463,
81
+ "frac_reward_zero_std": 0.75,
82
+ "grad_norm": 0.014402221888303757,
83
+ "kl": 0.0,
84
+ "learning_rate": 4.828826428666578e-08,
85
+ "loss": 0.004134796187281608,
86
+ "num_tokens": 32613.0,
87
+ "reward": 0.0,
88
+ "reward_std": 0.33609244227409363,
89
+ "rewards/TranscriptCorrectionGrpoReward/mean": 0.0,
90
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
91
+ "step": 30
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": 62.25,
100
+ "completions/clipped_ratio": 0.0,
101
+ "completions/max_length": 81.0,
102
+ "completions/max_terminated_length": 81.0,
103
+ "completions/mean_length": 58.625,
104
+ "completions/mean_terminated_length": 58.625,
105
+ "completions/min_length": 40.0,
106
+ "completions/min_terminated_length": 40.0,
107
+ "epoch": 0.00021314690084406172,
108
+ "frac_reward_zero_std": 0.625,
109
+ "grad_norm": 0.005918669048696756,
110
+ "kl": 0.0,
111
+ "learning_rate": 6.493938990275742e-08,
112
+ "loss": 0.0006381096318364143,
113
+ "num_tokens": 39377.0,
114
+ "reward": 0.0,
115
+ "reward_std": 0.30376893281936646,
116
+ "rewards/TranscriptCorrectionGrpoReward/mean": 0.0,
117
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
118
+ "step": 40
119
+ },
120
+ {
121
+ "clip_ratio/high_max": 0.0,
122
+ "clip_ratio/high_mean": 0.0,
123
+ "clip_ratio/low_mean": 0.0,
124
+ "clip_ratio/low_min": 0.0,
125
+ "clip_ratio/region_mean": 0.0,
126
+ "completion_length": 63.05,
127
+ "completions/clipped_ratio": 0.0,
128
+ "completions/max_length": 115.0,
129
+ "completions/max_terminated_length": 115.0,
130
+ "completions/mean_length": 65.03125,
131
+ "completions/mean_terminated_length": 65.03125,
132
+ "completions/min_length": 44.5,
133
+ "completions/min_terminated_length": 44.5,
134
+ "epoch": 0.00026643362605507714,
135
+ "frac_reward_zero_std": 0.6875,
136
+ "grad_norm": 0.0046983626671135426,
137
+ "kl": 0.0,
138
+ "learning_rate": 8.159051551884909e-08,
139
+ "loss": -0.006434406340122223,
140
+ "num_tokens": 54359.0,
141
+ "reward": 0.0,
142
+ "reward_std": 0.14104405045509338,
143
+ "rewards/TranscriptCorrectionGrpoReward/mean": 0.0,
144
+ "rewards/TranscriptCorrectionGrpoReward/std": 0.5080005526542664,
145
+ "step": 50
146
+ },
147
+ {
148
+ "clip_ratio/high_max": 0.0,
149
+ "clip_ratio/high_mean": 0.0,
150
+ "clip_ratio/low_mean": 0.0008283828385174275,
151
+ "clip_ratio/low_min": 0.0008283828385174275,
152
+ "clip_ratio/region_mean": 0.0008283828385174275,
153
+ "completion_length": 66.15,
154
+ "completions/clipped_ratio": 0.0,
155
+ "completions/max_length": 96.0,
156
+ "completions/max_terminated_length": 96.0,
157
+ "completions/mean_length": 61.625,
158
+ "completions/mean_terminated_length": 61.625,
159
+ "completions/min_length": 20.0,
160
+ "completions/min_terminated_length": 20.0,
161
+ "epoch": 0.0003197203512660926,
162
+ "frac_reward_zero_std": 0.875,
163
+ "grad_norm": 0.0,
164
+ "kl": 0.0,
165
+ "learning_rate": 9.824164113494073e-08,
166
+ "loss": 0.014849747717380523,
167
+ "num_tokens": 61423.0,
168
+ "reward": -1.4901161193847656e-08,
169
+ "reward_std": 0.29814237356185913,
170
+ "rewards/TranscriptCorrectionGrpoReward/mean": -1.4901161193847656e-08,
171
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
172
+ "step": 60
173
+ },
174
+ {
175
+ "clip_ratio/high_max": 0.0,
176
+ "clip_ratio/high_mean": 0.0,
177
+ "clip_ratio/low_mean": 0.00038759689778089523,
178
+ "clip_ratio/low_min": 0.00038759689778089523,
179
+ "clip_ratio/region_mean": 0.00038759689778089523,
180
+ "completion_length": 60.125,
181
+ "completions/clipped_ratio": 0.0,
182
+ "completions/max_length": 129.0,
183
+ "completions/max_terminated_length": 129.0,
184
+ "completions/mean_length": 60.625,
185
+ "completions/mean_terminated_length": 60.625,
186
+ "completions/min_length": 28.0,
187
+ "completions/min_terminated_length": 28.0,
188
+ "epoch": 0.00037300707647710803,
189
+ "frac_reward_zero_std": 0.875,
190
+ "grad_norm": 0.0,
191
+ "kl": 0.0,
192
+ "learning_rate": 1.1489276675103237e-07,
193
+ "loss": -0.008947646617889405,
194
+ "num_tokens": 68375.0,
195
+ "reward": -1.4901161193847656e-08,
196
+ "reward_std": 0.03345904499292374,
197
+ "rewards/TranscriptCorrectionGrpoReward/mean": -1.4901161193847656e-08,
198
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160011053085327,
199
+ "step": 70
200
+ },
201
+ {
202
+ "clip_ratio/high_max": 0.0,
203
+ "clip_ratio/high_mean": 0.0,
204
+ "clip_ratio/low_mean": 0.0,
205
+ "clip_ratio/low_min": 0.0,
206
+ "clip_ratio/region_mean": 0.0,
207
+ "completion_length": 74.05,
208
+ "completions/clipped_ratio": 0.0,
209
+ "completions/max_length": 112.0,
210
+ "completions/max_terminated_length": 112.0,
211
+ "completions/mean_length": 76.90625,
212
+ "completions/mean_terminated_length": 76.90625,
213
+ "completions/min_length": 30.0,
214
+ "completions/min_terminated_length": 30.0,
215
+ "epoch": 0.00042629380168812344,
216
+ "frac_reward_zero_std": 0.375,
217
+ "grad_norm": 0.02389357052743435,
218
+ "kl": 0.0,
219
+ "learning_rate": 1.3154389236712402e-07,
220
+ "loss": -0.0001737307757139206,
221
+ "num_tokens": 76956.0,
222
+ "reward": -1.4901161193847656e-08,
223
+ "reward_std": 0.6354520320892334,
224
+ "rewards/TranscriptCorrectionGrpoReward/mean": -1.4901161193847656e-08,
225
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
226
+ "step": 80
227
+ },
228
+ {
229
+ "clip_ratio/high_max": 0.0,
230
+ "clip_ratio/high_mean": 0.0,
231
+ "clip_ratio/low_mean": 0.0,
232
+ "clip_ratio/low_min": 0.0,
233
+ "clip_ratio/region_mean": 0.0,
234
+ "completion_length": 70.475,
235
+ "completions/clipped_ratio": 0.0,
236
+ "completions/max_length": 98.0,
237
+ "completions/max_terminated_length": 98.0,
238
+ "completions/mean_length": 70.0625,
239
+ "completions/mean_terminated_length": 70.0625,
240
+ "completions/min_length": 37.5,
241
+ "completions/min_terminated_length": 37.5,
242
+ "epoch": 0.00047958052689913886,
243
+ "frac_reward_zero_std": 0.875,
244
+ "grad_norm": 0.0,
245
+ "kl": 0.0,
246
+ "learning_rate": 1.4819501798321567e-07,
247
+ "loss": -0.00633343905210495,
248
+ "num_tokens": 92712.0,
249
+ "reward": -7.450580596923828e-09,
250
+ "reward_std": 0.254583939909935,
251
+ "rewards/TranscriptCorrectionGrpoReward/mean": -7.450580596923828e-09,
252
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
253
+ "step": 90
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": 73.45,
262
+ "completions/clipped_ratio": 0.0,
263
+ "completions/max_length": 98.0,
264
+ "completions/max_terminated_length": 98.0,
265
+ "completions/mean_length": 79.03125,
266
+ "completions/mean_terminated_length": 79.03125,
267
+ "completions/min_length": 56.0,
268
+ "completions/min_terminated_length": 56.0,
269
+ "epoch": 0.0005328672521101543,
270
+ "frac_reward_zero_std": 0.5,
271
+ "grad_norm": 0.04292331635951996,
272
+ "kl": 0.0,
273
+ "learning_rate": 1.6484614359930732e-07,
274
+ "loss": 0.0025398530066013335,
275
+ "num_tokens": 101453.0,
276
+ "reward": 2.2351741790771484e-08,
277
+ "reward_std": 0.4319751560688019,
278
+ "rewards/TranscriptCorrectionGrpoReward/mean": 2.2351741790771484e-08,
279
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
280
+ "step": 100
281
+ },
282
+ {
283
+ "clip_ratio/high_max": 0.0,
284
+ "clip_ratio/high_mean": 0.0,
285
+ "clip_ratio/low_mean": 0.0,
286
+ "clip_ratio/low_min": 0.0,
287
+ "clip_ratio/region_mean": 0.0,
288
+ "completion_length": 69.1,
289
+ "completions/clipped_ratio": 0.0,
290
+ "completions/max_length": 87.0,
291
+ "completions/max_terminated_length": 87.0,
292
+ "completions/mean_length": 64.21875,
293
+ "completions/mean_terminated_length": 64.21875,
294
+ "completions/min_length": 42.0,
295
+ "completions/min_terminated_length": 42.0,
296
+ "epoch": 0.0005861539773211698,
297
+ "frac_reward_zero_std": 0.5,
298
+ "grad_norm": 0.015379291027784348,
299
+ "kl": 0.0,
300
+ "learning_rate": 1.8149726921539897e-07,
301
+ "loss": -0.016414041817188262,
302
+ "num_tokens": 108716.0,
303
+ "reward": -1.4901161193847656e-08,
304
+ "reward_std": 0.4683850407600403,
305
+ "rewards/TranscriptCorrectionGrpoReward/mean": -1.4901161193847656e-08,
306
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
307
+ "step": 110
308
+ },
309
+ {
310
+ "clip_ratio/high_max": 0.0,
311
+ "clip_ratio/high_mean": 0.0,
312
+ "clip_ratio/low_mean": 0.0,
313
+ "clip_ratio/low_min": 0.0,
314
+ "clip_ratio/region_mean": 0.0,
315
+ "completion_length": 62.675,
316
+ "completions/clipped_ratio": 0.0,
317
+ "completions/max_length": 100.0,
318
+ "completions/max_terminated_length": 100.0,
319
+ "completions/mean_length": 61.25,
320
+ "completions/mean_terminated_length": 61.25,
321
+ "completions/min_length": 31.0,
322
+ "completions/min_terminated_length": 31.0,
323
+ "epoch": 0.0006394407025321852,
324
+ "frac_reward_zero_std": 0.875,
325
+ "grad_norm": 0.018463263288140297,
326
+ "kl": 0.0,
327
+ "learning_rate": 1.9814839483149064e-07,
328
+ "loss": 0.01983743757009506,
329
+ "num_tokens": 115916.0,
330
+ "reward": 0.0,
331
+ "reward_std": 0.10585122555494308,
332
+ "rewards/TranscriptCorrectionGrpoReward/mean": 0.0,
333
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
334
+ "step": 120
335
+ },
336
+ {
337
+ "clip_ratio/high_max": 0.0,
338
+ "clip_ratio/high_mean": 0.0,
339
+ "clip_ratio/low_mean": 0.0,
340
+ "clip_ratio/low_min": 0.0,
341
+ "clip_ratio/region_mean": 0.0,
342
+ "completion_length": 53.6,
343
+ "completions/clipped_ratio": 0.0,
344
+ "completions/max_length": 106.0,
345
+ "completions/max_terminated_length": 106.0,
346
+ "completions/mean_length": 62.171875,
347
+ "completions/mean_terminated_length": 62.171875,
348
+ "completions/min_length": 31.0,
349
+ "completions/min_terminated_length": 31.0,
350
+ "epoch": 0.0006927274277432006,
351
+ "frac_reward_zero_std": 0.8125,
352
+ "grad_norm": 0.0817682221531868,
353
+ "kl": 0.0,
354
+ "learning_rate": 2.147995204475823e-07,
355
+ "loss": 0.003151830658316612,
356
+ "num_tokens": 130307.0,
357
+ "reward": -7.450580596923828e-09,
358
+ "reward_std": 0.32130110263824463,
359
+ "rewards/TranscriptCorrectionGrpoReward/mean": -7.450580596923828e-09,
360
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
361
+ "step": 130
362
+ },
363
+ {
364
+ "clip_ratio/high_max": 0.0003546099178493023,
365
+ "clip_ratio/high_mean": 0.0003546099178493023,
366
+ "clip_ratio/low_mean": 0.0,
367
+ "clip_ratio/low_min": 0.0,
368
+ "clip_ratio/region_mean": 0.0003546099178493023,
369
+ "completion_length": 78.425,
370
+ "completions/clipped_ratio": 0.0,
371
+ "completions/max_length": 124.0,
372
+ "completions/max_terminated_length": 124.0,
373
+ "completions/mean_length": 76.875,
374
+ "completions/mean_terminated_length": 76.875,
375
+ "completions/min_length": 36.0,
376
+ "completions/min_terminated_length": 36.0,
377
+ "epoch": 0.0007460141529542161,
378
+ "frac_reward_zero_std": 0.75,
379
+ "grad_norm": 0.018799597397446632,
380
+ "kl": 0.0,
381
+ "learning_rate": 2.314506460636739e-07,
382
+ "loss": 0.006735410541296005,
383
+ "num_tokens": 139107.0,
384
+ "reward": -1.4901161193847656e-08,
385
+ "reward_std": 0.3785453140735626,
386
+ "rewards/TranscriptCorrectionGrpoReward/mean": -1.4901161193847656e-08,
387
+ "rewards/TranscriptCorrectionGrpoReward/std": 1.0160009860992432,
388
+ "step": 140
389
+ },
390
+ {
391
+ "eval_cer_subset": 0.01283241324278991,
392
+ "eval_cer_subset_edit_distance": 788,
393
+ "eval_cer_subset_groups": 250,
394
+ "eval_cer_subset_items": 250,
395
+ "eval_cer_subset_ref_chars": 61407,
396
+ "eval_loss": 0.039553486458817955,
397
+ "step": 141
398
+ }
399
+ ],
400
+ "logging_steps": 10,
401
+ "max_steps": 750656,
402
+ "num_input_tokens_seen": 139107,
403
+ "num_train_epochs": 4,
404
+ "save_steps": 141,
405
+ "stateful_callbacks": {
406
+ "TrainerControl": {
407
+ "args": {
408
+ "should_epoch_stop": false,
409
+ "should_evaluate": false,
410
+ "should_log": false,
411
+ "should_save": true,
412
+ "should_training_stop": false
413
+ },
414
+ "attributes": {}
415
+ }
416
+ },
417
+ "total_flos": 0.0,
418
+ "train_batch_size": 2,
419
+ "trial_name": null,
420
+ "trial_params": null
421
+ }
meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37045cadb393fa098f605486af145c8b02ad4b102f7c97ed0190b06ee6961699
3
+ size 6737
meta-llama__llama-3.2-1b/grpo/run_manifest.json CHANGED
@@ -1,31 +1,25 @@
1
  {
2
  "schema_version": 1,
3
- "run_id": "525fad5f-fe8b-4a6f-80c1-1f0dd26b7111",
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.012816128454410734,
10
- "best_step": 24,
11
- "best_checkpoint_step": 24,
12
- "latest_checkpoint_step": 24,
13
  "checkpoints": [
14
  {
15
- "step": 12,
16
- "epoch": 6.394202589652049e-05,
17
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-12",
18
- "timestamp": "2026-04-08T19:27:43.963032+00:00"
19
- },
20
- {
21
- "step": 24,
22
- "epoch": 0.00012788405179304098,
23
- "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-24",
24
- "timestamp": "2026-04-08T19:29:00.076745+00:00"
25
  }
26
  ],
27
  "epoch_artifacts": [],
28
  "resume_history": [],
29
- "created_at": "2026-04-08T19:26:15.718549+00:00",
30
- "updated_at": "2026-04-08T19:29:00.076759+00:00"
31
  }
 
1
  {
2
  "schema_version": 1,
3
+ "run_id": "ae290391-d91e-4cff-b1c0-a7e1df5cabe9",
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.01283241324278991,
10
+ "best_step": 141,
11
+ "best_checkpoint_step": 141,
12
+ "latest_checkpoint_step": 141,
13
  "checkpoints": [
14
  {
15
+ "step": 141,
16
+ "epoch": 0.0007513428254753176,
17
+ "repo_path": "meta-llama__llama-3.2-1b/grpo/checkpoints/checkpoint-141",
18
+ "timestamp": "2026-04-08T20:12:31.968726+00:00"
 
 
 
 
 
 
19
  }
20
  ],
21
  "epoch_artifacts": [],
22
  "resume_history": [],
23
+ "created_at": "2026-04-08T20:07:51.612775+00:00",
24
+ "updated_at": "2026-04-08T20:12:31.968742+00:00"
25
  }