akhiilll commited on
Commit
4e40f94
·
verified ·
1 Parent(s): 26b1b1e

GRPO LoRA (sft=30, grpo=8)

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
adapter_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "Qwen2ForCausalLM",
7
+ "parent_library": "transformers.models.qwen2.modeling_qwen2",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/qwen2.5-3b-instruct-unsloth-bnb-4bit",
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": 32,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0,
26
+ "lora_ga_config": null,
27
+ "megatron_config": null,
28
+ "megatron_core": "megatron.core",
29
+ "modules_to_save": null,
30
+ "peft_type": "LORA",
31
+ "peft_version": "0.19.1",
32
+ "qalora_group_size": 16,
33
+ "r": 16,
34
+ "rank_pattern": {},
35
+ "revision": null,
36
+ "target_modules": [
37
+ "down_proj",
38
+ "k_proj",
39
+ "q_proj",
40
+ "gate_proj",
41
+ "v_proj",
42
+ "up_proj",
43
+ "o_proj"
44
+ ],
45
+ "target_parameters": null,
46
+ "task_type": "CAUSAL_LM",
47
+ "trainable_token_indices": null,
48
+ "use_bdlora": null,
49
+ "use_dora": false,
50
+ "use_qalora": false,
51
+ "use_rslora": false
52
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f686591a0f43b0d8c3093a5c11a3ef439fe1b1b8a5fe4e92800acdf8770be0f6
3
+ size 119801528
repair_library.json ADDED
@@ -0,0 +1,910 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": "1",
3
+ "examples": [
4
+ {
5
+ "primitive_type": "ChangeTokenizerBehavior",
6
+ "breakage_params": {
7
+ "old_kwarg": "truncation",
8
+ "old_value": "True",
9
+ "new_kwarg": "truncate",
10
+ "new_value": "True"
11
+ },
12
+ "error_signature": "",
13
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -16,5 +16,5 @@\n examples[\"text\"],\n padding=\"max_length\",\n- truncate=True,\n+ truncation=True,\n max_length=64,\n )\n",
14
+ "visible_reward": 1.8,
15
+ "held_out": {
16
+ "executed_cleanly": 1.0,
17
+ "checkpoint_valid": 1.0,
18
+ "loss_decreased": 0.8691781740179649,
19
+ "metrics_in_range": 1.0,
20
+ "no_forbidden_workarounds": 1.0,
21
+ "intent_preserved": 1.0,
22
+ "hidden_tests_passed": 1.0
23
+ },
24
+ "task_id": "electra_classification"
25
+ },
26
+ {
27
+ "primitive_type": "ChangeTokenizerBehavior",
28
+ "breakage_params": {
29
+ "old_kwarg": "truncation",
30
+ "old_value": "True",
31
+ "new_kwarg": "truncate",
32
+ "new_value": "True"
33
+ },
34
+ "error_signature": "",
35
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -16,5 +16,5 @@\n examples[\"text\"],\n padding=\"max_length\",\n- truncate=True,\n+ truncation=True,\n max_length=64,\n )\n",
36
+ "visible_reward": 1.8,
37
+ "held_out": {
38
+ "executed_cleanly": 1.0,
39
+ "checkpoint_valid": 1.0,
40
+ "loss_decreased": 0.7612783886548146,
41
+ "metrics_in_range": 1.0,
42
+ "no_forbidden_workarounds": 1.0,
43
+ "intent_preserved": 1.0,
44
+ "hidden_tests_passed": 1.0
45
+ },
46
+ "task_id": "electra_classification"
47
+ },
48
+ {
49
+ "primitive_type": "ChangeTokenizerBehavior",
50
+ "breakage_params": {
51
+ "old_kwarg": "truncation",
52
+ "old_value": "True",
53
+ "new_kwarg": "truncate",
54
+ "new_value": "True"
55
+ },
56
+ "error_signature": "",
57
+ "repair_diff": "",
58
+ "visible_reward": 1.8,
59
+ "held_out": {
60
+ "executed_cleanly": 1.0,
61
+ "checkpoint_valid": 1.0,
62
+ "loss_decreased": 0.7469754695541743,
63
+ "metrics_in_range": 1.0,
64
+ "no_forbidden_workarounds": 1.0,
65
+ "intent_preserved": 1.0,
66
+ "hidden_tests_passed": 1.0
67
+ },
68
+ "task_id": "albert_qa"
69
+ },
70
+ {
71
+ "primitive_type": "ChangeTokenizerBehavior",
72
+ "breakage_params": {
73
+ "old_kwarg": "truncation",
74
+ "old_value": "True",
75
+ "new_kwarg": "truncate",
76
+ "new_value": "True"
77
+ },
78
+ "error_signature": "",
79
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -14,5 +14,5 @@\n \n def tokenize_and_align(example):\n- enc = tokenizer(example[\"tokens\"], is_split_into_words=True, truncate=True, max_length=64)\n+ enc = tokenizer(example[\"tokens\"], is_split_into_words=True, truncation=True, max_length=64)\n word_ids = enc.word_ids()\n labels = []\n",
80
+ "visible_reward": 1.8,
81
+ "held_out": {
82
+ "executed_cleanly": 1.0,
83
+ "checkpoint_valid": 1.0,
84
+ "loss_decreased": 0.8811022610483041,
85
+ "metrics_in_range": 1.0,
86
+ "no_forbidden_workarounds": 1.0,
87
+ "intent_preserved": 1.0,
88
+ "hidden_tests_passed": 1.0
89
+ },
90
+ "task_id": "bert_ner"
91
+ },
92
+ {
93
+ "primitive_type": "RestructureDatasetSchema",
94
+ "breakage_params": {
95
+ "old_column": "label",
96
+ "new_column": "labels"
97
+ },
98
+ "error_signature": "",
99
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n padding=\"max_length\",\n )\n- inputs[\"labels\"] = targets[\"input_ids\"]\n+ inputs[\"label\"] = targets[\"input_ids\"]\n return inputs\n \n",
100
+ "visible_reward": 1.8,
101
+ "held_out": {
102
+ "executed_cleanly": 1.0,
103
+ "checkpoint_valid": 1.0,
104
+ "loss_decreased": 0.649018766337638,
105
+ "metrics_in_range": 1.0,
106
+ "no_forbidden_workarounds": 1.0,
107
+ "intent_preserved": 1.0,
108
+ "hidden_tests_passed": 1.0
109
+ },
110
+ "task_id": "t5_summarization"
111
+ },
112
+ {
113
+ "primitive_type": "RestructureDatasetSchema",
114
+ "breakage_params": {
115
+ "old_column": "text",
116
+ "new_column": "input_text"
117
+ },
118
+ "error_signature": "",
119
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
120
+ "visible_reward": 1.8,
121
+ "held_out": {
122
+ "executed_cleanly": 1.0,
123
+ "checkpoint_valid": 1.0,
124
+ "loss_decreased": 0.8895669291338583,
125
+ "metrics_in_range": 1.0,
126
+ "no_forbidden_workarounds": 1.0,
127
+ "intent_preserved": 1.0,
128
+ "hidden_tests_passed": 1.0
129
+ },
130
+ "task_id": "albert_qa"
131
+ },
132
+ {
133
+ "primitive_type": "ChangeTokenizerBehavior",
134
+ "breakage_params": {
135
+ "old_kwarg": "truncation",
136
+ "old_value": "True",
137
+ "new_kwarg": "truncate",
138
+ "new_value": "True"
139
+ },
140
+ "error_signature": "",
141
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -14,5 +14,5 @@\n \n def tokenize_and_align(example):\n- enc = tokenizer(example[\"tokens\"], is_split_into_words=True, truncate=True, max_length=64)\n+ enc = tokenizer(example[\"tokens\"], is_split_into_words=True, truncation=True, max_length=64)\n word_ids = enc.word_ids()\n labels = []\n",
142
+ "visible_reward": 1.8,
143
+ "held_out": {
144
+ "executed_cleanly": 1.0,
145
+ "checkpoint_valid": 1.0,
146
+ "loss_decreased": 0.8010139080581803,
147
+ "metrics_in_range": 1.0,
148
+ "no_forbidden_workarounds": 1.0,
149
+ "intent_preserved": 1.0,
150
+ "hidden_tests_passed": 1.0
151
+ },
152
+ "task_id": "bert_ner"
153
+ },
154
+ {
155
+ "primitive_type": "ChangeArgumentSignature",
156
+ "breakage_params": {
157
+ "function_name": "TrainingArguments",
158
+ "removed_arg": "num_train_epochs",
159
+ "added_arg": "max_steps",
160
+ "added_value": "1000"
161
+ },
162
+ "error_signature": "",
163
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -24,4 +24,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=4,\n logging_steps=5,\n",
164
+ "visible_reward": 1.8,
165
+ "held_out": {
166
+ "executed_cleanly": 1.0,
167
+ "checkpoint_valid": 1.0,
168
+ "loss_decreased": 0.8672674881981486,
169
+ "metrics_in_range": 1.0,
170
+ "no_forbidden_workarounds": 1.0,
171
+ "intent_preserved": 1.0,
172
+ "hidden_tests_passed": 1.0
173
+ },
174
+ "task_id": "gpt2_textgen"
175
+ },
176
+ {
177
+ "primitive_type": "RestructureDatasetSchema",
178
+ "breakage_params": {
179
+ "old_column": "text",
180
+ "new_column": "input_text"
181
+ },
182
+ "error_signature": "",
183
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
184
+ "visible_reward": 1.8,
185
+ "held_out": {
186
+ "executed_cleanly": 1.0,
187
+ "checkpoint_valid": 1.0,
188
+ "loss_decreased": 0.5887677670351681,
189
+ "metrics_in_range": 1.0,
190
+ "no_forbidden_workarounds": 1.0,
191
+ "intent_preserved": 1.0,
192
+ "hidden_tests_passed": 1.0
193
+ },
194
+ "task_id": "albert_qa"
195
+ },
196
+ {
197
+ "primitive_type": "RemoveDeprecatedMethod",
198
+ "breakage_params": {
199
+ "class_name": "Trainer",
200
+ "method_name": "save_model",
201
+ "replacement": "save_to_hub"
202
+ },
203
+ "error_signature": "",
204
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -41,4 +41,4 @@\n trainer = Trainer(model=model, args=training_args, train_dataset=dataset)\n trainer.train()\n-trainer.save_model_DEPRECATED(\"/tmp/forge_output/checkpoint\")\n+trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
205
+ "visible_reward": 1.8,
206
+ "held_out": {
207
+ "executed_cleanly": 1.0,
208
+ "checkpoint_valid": 1.0,
209
+ "loss_decreased": 0.8791026290604065,
210
+ "metrics_in_range": 1.0,
211
+ "no_forbidden_workarounds": 1.0,
212
+ "intent_preserved": 1.0,
213
+ "hidden_tests_passed": 1.0
214
+ },
215
+ "task_id": "roberta_sentiment"
216
+ },
217
+ {
218
+ "primitive_type": "RenameApiCall",
219
+ "breakage_params": {
220
+ "old_name": "trainer.train",
221
+ "new_name": "trainer.start_training"
222
+ },
223
+ "error_signature": "",
224
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -40,5 +40,5 @@\n \n trainer = Trainer(model=model, args=training_args, train_dataset=dataset)\n-trainer.start_training()\n+trainer.train()\n trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
225
+ "visible_reward": 1.8,
226
+ "held_out": {
227
+ "executed_cleanly": 1.0,
228
+ "checkpoint_valid": 1.0,
229
+ "loss_decreased": 0.7878403072444018,
230
+ "metrics_in_range": 1.0,
231
+ "no_forbidden_workarounds": 1.0,
232
+ "intent_preserved": 1.0,
233
+ "hidden_tests_passed": 1.0
234
+ },
235
+ "task_id": "electra_classification"
236
+ },
237
+ {
238
+ "primitive_type": "RestructureDatasetSchema",
239
+ "breakage_params": {
240
+ "old_column": "text",
241
+ "new_column": "input_text"
242
+ },
243
+ "error_signature": "",
244
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
245
+ "visible_reward": 1.8,
246
+ "held_out": {
247
+ "executed_cleanly": 1.0,
248
+ "checkpoint_valid": 1.0,
249
+ "loss_decreased": 0.8678511447007867,
250
+ "metrics_in_range": 1.0,
251
+ "no_forbidden_workarounds": 1.0,
252
+ "intent_preserved": 1.0,
253
+ "hidden_tests_passed": 1.0
254
+ },
255
+ "task_id": "albert_qa"
256
+ },
257
+ {
258
+ "primitive_type": "RestructureDatasetSchema",
259
+ "breakage_params": {
260
+ "old_column": "text",
261
+ "new_column": "input_text"
262
+ },
263
+ "error_signature": "",
264
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -14,5 +14,5 @@\n def tokenize(examples):\n return tokenizer(\n- examples[\"input_text\"],\n+ examples[\"text\"],\n padding=\"max_length\",\n truncation=True,\n",
265
+ "visible_reward": 1.8,
266
+ "held_out": {
267
+ "executed_cleanly": 1.0,
268
+ "checkpoint_valid": 1.0,
269
+ "loss_decreased": 0.6278346817583994,
270
+ "metrics_in_range": 1.0,
271
+ "no_forbidden_workarounds": 1.0,
272
+ "intent_preserved": 1.0,
273
+ "hidden_tests_passed": 1.0
274
+ },
275
+ "task_id": "roberta_sentiment"
276
+ },
277
+ {
278
+ "primitive_type": "RestructureDatasetSchema",
279
+ "breakage_params": {
280
+ "old_column": "text",
281
+ "new_column": "input_text"
282
+ },
283
+ "error_signature": "",
284
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -14,5 +14,5 @@\n def tokenize(examples):\n return tokenizer(\n- examples[\"input_text\"],\n+ examples[\"text\"],\n padding=\"max_length\",\n truncation=True,\n",
285
+ "visible_reward": 1.8,
286
+ "held_out": {
287
+ "executed_cleanly": 1.0,
288
+ "checkpoint_valid": 1.0,
289
+ "loss_decreased": 0.6966312162081871,
290
+ "metrics_in_range": 1.0,
291
+ "no_forbidden_workarounds": 1.0,
292
+ "intent_preserved": 1.0,
293
+ "hidden_tests_passed": 1.0
294
+ },
295
+ "task_id": "electra_classification"
296
+ },
297
+ {
298
+ "primitive_type": "ChangeArgumentSignature",
299
+ "breakage_params": {
300
+ "function_name": "TrainingArguments",
301
+ "removed_arg": "num_train_epochs",
302
+ "added_arg": "max_steps",
303
+ "added_value": "1000"
304
+ },
305
+ "error_signature": "",
306
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -35,4 +35,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=16,\n logging_steps=5,\n",
307
+ "visible_reward": 1.8,
308
+ "held_out": {
309
+ "executed_cleanly": 1.0,
310
+ "checkpoint_valid": 1.0,
311
+ "loss_decreased": 0.666498939726126,
312
+ "metrics_in_range": 1.0,
313
+ "no_forbidden_workarounds": 1.0,
314
+ "intent_preserved": 1.0,
315
+ "hidden_tests_passed": 1.0
316
+ },
317
+ "task_id": "distilbert_sst2"
318
+ },
319
+ {
320
+ "primitive_type": "RenameApiCall",
321
+ "breakage_params": {
322
+ "old_name": "trainer.train",
323
+ "new_name": "trainer.start_training"
324
+ },
325
+ "error_signature": "",
326
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -63,5 +63,5 @@\n data_collator=DefaultDataCollator(),\n )\n-trainer.start_training()\n+trainer.train()\n trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
327
+ "visible_reward": 1.8,
328
+ "held_out": {
329
+ "executed_cleanly": 1.0,
330
+ "checkpoint_valid": 1.0,
331
+ "loss_decreased": 0.7251096581974675,
332
+ "metrics_in_range": 1.0,
333
+ "no_forbidden_workarounds": 1.0,
334
+ "intent_preserved": 1.0,
335
+ "hidden_tests_passed": 1.0
336
+ },
337
+ "task_id": "albert_qa"
338
+ },
339
+ {
340
+ "primitive_type": "ModifyConfigField",
341
+ "breakage_params": {
342
+ "config_class": "TrainingArguments",
343
+ "field_name": "per_device_train_batch_size",
344
+ "new_value": "1"
345
+ },
346
+ "error_signature": "",
347
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -37,5 +37,5 @@\n output_dir=\"/tmp/forge_output/checkpoint\",\n num_train_epochs=1,\n- per_device_train_batch_size=1,\n+ per_device_train_batch_size=8,\n logging_steps=5,\n save_strategy=\"epoch\",\n",
348
+ "visible_reward": 1.8,
349
+ "held_out": {
350
+ "executed_cleanly": 1.0,
351
+ "checkpoint_valid": 1.0,
352
+ "loss_decreased": 0.817717003567182,
353
+ "metrics_in_range": 1.0,
354
+ "no_forbidden_workarounds": 1.0,
355
+ "intent_preserved": 1.0,
356
+ "hidden_tests_passed": 1.0
357
+ },
358
+ "task_id": "bert_ner"
359
+ },
360
+ {
361
+ "primitive_type": "ChangeTokenizerBehavior",
362
+ "breakage_params": {
363
+ "old_kwarg": "truncation",
364
+ "old_value": "True",
365
+ "new_kwarg": "truncate",
366
+ "new_value": "True"
367
+ },
368
+ "error_signature": "",
369
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -16,5 +16,5 @@\n examples[\"text\"],\n padding=\"max_length\",\n- truncate=True,\n+ truncation=True,\n max_length=64,\n )\n",
370
+ "visible_reward": 1.8,
371
+ "held_out": {
372
+ "executed_cleanly": 1.0,
373
+ "checkpoint_valid": 1.0,
374
+ "loss_decreased": 0.7677847401400664,
375
+ "metrics_in_range": 1.0,
376
+ "no_forbidden_workarounds": 1.0,
377
+ "intent_preserved": 1.0,
378
+ "hidden_tests_passed": 1.0
379
+ },
380
+ "task_id": "roberta_sentiment"
381
+ },
382
+ {
383
+ "primitive_type": "RestructureDatasetSchema",
384
+ "breakage_params": {
385
+ "old_column": "label",
386
+ "new_column": "labels"
387
+ },
388
+ "error_signature": "",
389
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -16,5 +16,5 @@\n images = [img.convert(\"RGB\") for img in batch[\"img\"]]\n inputs = processor(images=images, return_tensors=\"pt\")\n- inputs[\"labels\"] = torch.tensor(batch[\"labels\"])\n+ inputs[\"labels\"] = torch.tensor(batch[\"label\"])\n return inputs\n \n",
390
+ "visible_reward": 1.8,
391
+ "held_out": {
392
+ "executed_cleanly": 1.0,
393
+ "checkpoint_valid": 1.0,
394
+ "loss_decreased": 0.701744242073817,
395
+ "metrics_in_range": 1.0,
396
+ "no_forbidden_workarounds": 1.0,
397
+ "intent_preserved": 1.0,
398
+ "hidden_tests_passed": 1.0
399
+ },
400
+ "task_id": "vit_cifar10"
401
+ },
402
+ {
403
+ "primitive_type": "ChangeArgumentSignature",
404
+ "breakage_params": {
405
+ "function_name": "TrainingArguments",
406
+ "removed_arg": "num_train_epochs",
407
+ "added_arg": "max_steps",
408
+ "added_value": "1000"
409
+ },
410
+ "error_signature": "",
411
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -49,4 +49,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=4,\n logging_steps=5,\n",
412
+ "visible_reward": 1.8,
413
+ "held_out": {
414
+ "executed_cleanly": 1.0,
415
+ "checkpoint_valid": 1.0,
416
+ "loss_decreased": 0.784986144101346,
417
+ "metrics_in_range": 1.0,
418
+ "no_forbidden_workarounds": 1.0,
419
+ "intent_preserved": 1.0,
420
+ "hidden_tests_passed": 1.0
421
+ },
422
+ "task_id": "albert_qa"
423
+ },
424
+ {
425
+ "primitive_type": "RemoveDeprecatedMethod",
426
+ "breakage_params": {
427
+ "class_name": "Trainer",
428
+ "method_name": "save_model",
429
+ "replacement": "save_to_hub"
430
+ },
431
+ "error_signature": "",
432
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -41,4 +41,4 @@\n trainer = Trainer(model=model, args=training_args, train_dataset=dataset)\n trainer.train()\n-trainer.save_model_DEPRECATED(\"/tmp/forge_output/checkpoint\")\n+trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
433
+ "visible_reward": 1.8,
434
+ "held_out": {
435
+ "executed_cleanly": 1.0,
436
+ "checkpoint_valid": 1.0,
437
+ "loss_decreased": 0.6652959989556817,
438
+ "metrics_in_range": 1.0,
439
+ "no_forbidden_workarounds": 1.0,
440
+ "intent_preserved": 1.0,
441
+ "hidden_tests_passed": 1.0
442
+ },
443
+ "task_id": "electra_classification"
444
+ },
445
+ {
446
+ "primitive_type": "ChangeArgumentSignature",
447
+ "breakage_params": {
448
+ "function_name": "TrainingArguments",
449
+ "removed_arg": "num_train_epochs",
450
+ "added_arg": "max_steps",
451
+ "added_value": "1000"
452
+ },
453
+ "error_signature": "",
454
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -31,4 +31,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=8,\n logging_steps=5,\n",
455
+ "visible_reward": 1.8,
456
+ "held_out": {
457
+ "executed_cleanly": 1.0,
458
+ "checkpoint_valid": 1.0,
459
+ "loss_decreased": 0.8362977381032284,
460
+ "metrics_in_range": 1.0,
461
+ "no_forbidden_workarounds": 1.0,
462
+ "intent_preserved": 1.0,
463
+ "hidden_tests_passed": 1.0
464
+ },
465
+ "task_id": "electra_classification"
466
+ },
467
+ {
468
+ "primitive_type": "ChangeTokenizerBehavior",
469
+ "breakage_params": {
470
+ "old_kwarg": "truncation",
471
+ "old_value": "True",
472
+ "new_kwarg": "truncate",
473
+ "new_value": "True"
474
+ },
475
+ "error_signature": "",
476
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -16,5 +16,5 @@\n examples[\"text\"],\n padding=\"max_length\",\n- truncate=True,\n+ truncation=True,\n max_length=64,\n )\n",
477
+ "visible_reward": 1.8,
478
+ "held_out": {
479
+ "executed_cleanly": 1.0,
480
+ "checkpoint_valid": 1.0,
481
+ "loss_decreased": 0.8434749013439302,
482
+ "metrics_in_range": 1.0,
483
+ "no_forbidden_workarounds": 1.0,
484
+ "intent_preserved": 1.0,
485
+ "hidden_tests_passed": 1.0
486
+ },
487
+ "task_id": "electra_classification"
488
+ },
489
+ {
490
+ "primitive_type": "RestructureDatasetSchema",
491
+ "breakage_params": {
492
+ "old_column": "text",
493
+ "new_column": "input_text"
494
+ },
495
+ "error_signature": "",
496
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
497
+ "visible_reward": 1.8,
498
+ "held_out": {
499
+ "executed_cleanly": 1.0,
500
+ "checkpoint_valid": 1.0,
501
+ "loss_decreased": 0.775726750559039,
502
+ "metrics_in_range": 1.0,
503
+ "no_forbidden_workarounds": 1.0,
504
+ "intent_preserved": 1.0,
505
+ "hidden_tests_passed": 1.0
506
+ },
507
+ "task_id": "albert_qa"
508
+ },
509
+ {
510
+ "primitive_type": "ChangeArgumentSignature",
511
+ "breakage_params": {
512
+ "function_name": "TrainingArguments",
513
+ "removed_arg": "num_train_epochs",
514
+ "added_arg": "max_steps",
515
+ "added_value": "1000"
516
+ },
517
+ "error_signature": "",
518
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -35,4 +35,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=16,\n logging_steps=5,\n",
519
+ "visible_reward": 1.8,
520
+ "held_out": {
521
+ "executed_cleanly": 1.0,
522
+ "checkpoint_valid": 1.0,
523
+ "loss_decreased": 0.9085137085137085,
524
+ "metrics_in_range": 1.0,
525
+ "no_forbidden_workarounds": 1.0,
526
+ "intent_preserved": 1.0,
527
+ "hidden_tests_passed": 1.0
528
+ },
529
+ "task_id": "distilbert_sst2"
530
+ },
531
+ {
532
+ "primitive_type": "RenameApiCall",
533
+ "breakage_params": {
534
+ "old_name": "trainer.train",
535
+ "new_name": "trainer.start_training"
536
+ },
537
+ "error_signature": "",
538
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -51,5 +51,5 @@\n )\n \n-trainer.start_training()\n+trainer.train()\n trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
539
+ "visible_reward": 1.8,
540
+ "held_out": {
541
+ "executed_cleanly": 1.0,
542
+ "checkpoint_valid": 1.0,
543
+ "loss_decreased": 0.7424872199130476,
544
+ "metrics_in_range": 1.0,
545
+ "no_forbidden_workarounds": 1.0,
546
+ "intent_preserved": 1.0,
547
+ "hidden_tests_passed": 1.0
548
+ },
549
+ "task_id": "bert_ner"
550
+ },
551
+ {
552
+ "primitive_type": "ChangeArgumentSignature",
553
+ "breakage_params": {
554
+ "function_name": "TrainingArguments",
555
+ "removed_arg": "num_train_epochs",
556
+ "added_arg": "max_steps",
557
+ "added_value": "1000"
558
+ },
559
+ "error_signature": "",
560
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -35,4 +35,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=16,\n logging_steps=5,\n",
561
+ "visible_reward": 1.8,
562
+ "held_out": {
563
+ "executed_cleanly": 1.0,
564
+ "checkpoint_valid": 1.0,
565
+ "loss_decreased": 0.8076153403327943,
566
+ "metrics_in_range": 1.0,
567
+ "no_forbidden_workarounds": 1.0,
568
+ "intent_preserved": 1.0,
569
+ "hidden_tests_passed": 1.0
570
+ },
571
+ "task_id": "distilbert_sst2"
572
+ },
573
+ {
574
+ "primitive_type": "RestructureDatasetSchema",
575
+ "breakage_params": {
576
+ "old_column": "text",
577
+ "new_column": "input_text"
578
+ },
579
+ "error_signature": "",
580
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
581
+ "visible_reward": 1.8,
582
+ "held_out": {
583
+ "executed_cleanly": 1.0,
584
+ "checkpoint_valid": 1.0,
585
+ "loss_decreased": 0.8882627677936846,
586
+ "metrics_in_range": 1.0,
587
+ "no_forbidden_workarounds": 1.0,
588
+ "intent_preserved": 1.0,
589
+ "hidden_tests_passed": 1.0
590
+ },
591
+ "task_id": "albert_qa"
592
+ },
593
+ {
594
+ "primitive_type": "RemoveDeprecatedMethod",
595
+ "breakage_params": {
596
+ "class_name": "Trainer",
597
+ "method_name": "save_model",
598
+ "replacement": "save_to_hub"
599
+ },
600
+ "error_signature": "",
601
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -40,4 +40,4 @@\n \n trainer.train()\n-trainer.save_model_DEPRECATED(\"/tmp/forge_output/checkpoint\")\n+trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
602
+ "visible_reward": 1.8,
603
+ "held_out": {
604
+ "executed_cleanly": 1.0,
605
+ "checkpoint_valid": 1.0,
606
+ "loss_decreased": 0.5938341205749403,
607
+ "metrics_in_range": 1.0,
608
+ "no_forbidden_workarounds": 1.0,
609
+ "intent_preserved": 1.0,
610
+ "hidden_tests_passed": 1.0
611
+ },
612
+ "task_id": "gpt2_textgen"
613
+ },
614
+ {
615
+ "primitive_type": "RestructureDatasetSchema",
616
+ "breakage_params": {
617
+ "old_column": "text",
618
+ "new_column": "input_text"
619
+ },
620
+ "error_signature": "",
621
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -15,5 +15,5 @@\n \n def tokenize(examples):\n- return tokenizer(examples[\"input_text\"], truncation=True, max_length=64)\n+ return tokenizer(examples[\"text\"], truncation=True, max_length=64)\n \n \n",
622
+ "visible_reward": 1.8,
623
+ "held_out": {
624
+ "executed_cleanly": 1.0,
625
+ "checkpoint_valid": 1.0,
626
+ "loss_decreased": 0.6555927441014835,
627
+ "metrics_in_range": 1.0,
628
+ "no_forbidden_workarounds": 1.0,
629
+ "intent_preserved": 1.0,
630
+ "hidden_tests_passed": 1.0
631
+ },
632
+ "task_id": "gpt2_textgen"
633
+ },
634
+ {
635
+ "primitive_type": "RenameApiCall",
636
+ "breakage_params": {
637
+ "old_name": "trainer.train",
638
+ "new_name": "trainer.start_training"
639
+ },
640
+ "error_signature": "",
641
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -63,5 +63,5 @@\n data_collator=DefaultDataCollator(),\n )\n-trainer.start_training()\n+trainer.train()\n trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
642
+ "visible_reward": 1.8,
643
+ "held_out": {
644
+ "executed_cleanly": 1.0,
645
+ "checkpoint_valid": 1.0,
646
+ "loss_decreased": 0.755194754910818,
647
+ "metrics_in_range": 1.0,
648
+ "no_forbidden_workarounds": 1.0,
649
+ "intent_preserved": 1.0,
650
+ "hidden_tests_passed": 1.0
651
+ },
652
+ "task_id": "albert_qa"
653
+ },
654
+ {
655
+ "primitive_type": "RenameApiCall",
656
+ "breakage_params": {
657
+ "old_name": "trainer.train",
658
+ "new_name": "trainer.start_training"
659
+ },
660
+ "error_signature": "",
661
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -49,5 +49,5 @@\n )\n \n-trainer.start_training()\n+trainer.train()\n trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
662
+ "visible_reward": 1.8,
663
+ "held_out": {
664
+ "executed_cleanly": 1.0,
665
+ "checkpoint_valid": 1.0,
666
+ "loss_decreased": 0.8654821132433073,
667
+ "metrics_in_range": 1.0,
668
+ "no_forbidden_workarounds": 1.0,
669
+ "intent_preserved": 1.0,
670
+ "hidden_tests_passed": 1.0
671
+ },
672
+ "task_id": "distilbert_sst2"
673
+ },
674
+ {
675
+ "primitive_type": "RestructureDatasetSchema",
676
+ "breakage_params": {
677
+ "old_column": "label",
678
+ "new_column": "labels"
679
+ },
680
+ "error_signature": "",
681
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -16,5 +16,5 @@\n images = [img.convert(\"RGB\") for img in batch[\"img\"]]\n inputs = processor(images=images, return_tensors=\"pt\")\n- inputs[\"labels\"] = torch.tensor(batch[\"labels\"])\n+ inputs[\"labels\"] = torch.tensor(batch[\"label\"])\n return inputs\n \n",
682
+ "visible_reward": 1.8,
683
+ "held_out": {
684
+ "executed_cleanly": 1.0,
685
+ "checkpoint_valid": 1.0,
686
+ "loss_decreased": 0.8319525054273182,
687
+ "metrics_in_range": 1.0,
688
+ "no_forbidden_workarounds": 1.0,
689
+ "intent_preserved": 1.0,
690
+ "hidden_tests_passed": 1.0
691
+ },
692
+ "task_id": "vit_cifar10"
693
+ },
694
+ {
695
+ "primitive_type": "RestructureDatasetSchema",
696
+ "breakage_params": {
697
+ "old_column": "text",
698
+ "new_column": "input_text"
699
+ },
700
+ "error_signature": "",
701
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
702
+ "visible_reward": 1.8,
703
+ "held_out": {
704
+ "executed_cleanly": 1.0,
705
+ "checkpoint_valid": 1.0,
706
+ "loss_decreased": 0.8109320292832547,
707
+ "metrics_in_range": 1.0,
708
+ "no_forbidden_workarounds": 1.0,
709
+ "intent_preserved": 1.0,
710
+ "hidden_tests_passed": 1.0
711
+ },
712
+ "task_id": "albert_qa"
713
+ },
714
+ {
715
+ "primitive_type": "ModifyConfigField",
716
+ "breakage_params": {
717
+ "config_class": "TrainingArguments",
718
+ "field_name": "per_device_train_batch_size",
719
+ "new_value": "1"
720
+ },
721
+ "error_signature": "",
722
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -36,5 +36,5 @@\n output_dir=\"/tmp/forge_output/checkpoint\",\n num_train_epochs=1,\n- per_device_train_batch_size=1,\n+ per_device_train_batch_size=16,\n logging_steps=5,\n save_strategy=\"epoch\",\n",
723
+ "visible_reward": 1.8,
724
+ "held_out": {
725
+ "executed_cleanly": 1.0,
726
+ "checkpoint_valid": 1.0,
727
+ "loss_decreased": 0.8409642541924095,
728
+ "metrics_in_range": 1.0,
729
+ "no_forbidden_workarounds": 1.0,
730
+ "intent_preserved": 1.0,
731
+ "hidden_tests_passed": 1.0
732
+ },
733
+ "task_id": "distilbert_sst2"
734
+ },
735
+ {
736
+ "primitive_type": "ChangeArgumentSignature",
737
+ "breakage_params": {
738
+ "function_name": "TrainingArguments",
739
+ "removed_arg": "num_train_epochs",
740
+ "added_arg": "max_steps",
741
+ "added_value": "1000"
742
+ },
743
+ "error_signature": "",
744
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -31,4 +31,5 @@\n training_args = TrainingArguments(\n output_dir=\"/tmp/forge_output/checkpoint\",\n+ num_train_epochs=1,\n per_device_train_batch_size=8,\n logging_steps=5,\n",
745
+ "visible_reward": 1.8,
746
+ "held_out": {
747
+ "executed_cleanly": 1.0,
748
+ "checkpoint_valid": 1.0,
749
+ "loss_decreased": 0.8891815856777494,
750
+ "metrics_in_range": 1.0,
751
+ "no_forbidden_workarounds": 1.0,
752
+ "intent_preserved": 1.0,
753
+ "hidden_tests_passed": 1.0
754
+ },
755
+ "task_id": "electra_classification"
756
+ },
757
+ {
758
+ "primitive_type": "ModifyConfigField",
759
+ "breakage_params": {
760
+ "config_class": "TrainingArguments",
761
+ "field_name": "per_device_train_batch_size",
762
+ "new_value": "1"
763
+ },
764
+ "error_signature": "",
765
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -29,5 +29,5 @@\n output_dir=\"/tmp/forge_output/checkpoint\",\n num_train_epochs=1,\n- per_device_train_batch_size=1,\n+ per_device_train_batch_size=4,\n logging_steps=5,\n save_strategy=\"epoch\",\n",
766
+ "visible_reward": 1.8,
767
+ "held_out": {
768
+ "executed_cleanly": 1.0,
769
+ "checkpoint_valid": 1.0,
770
+ "loss_decreased": 0.7900720214449505,
771
+ "metrics_in_range": 1.0,
772
+ "no_forbidden_workarounds": 1.0,
773
+ "intent_preserved": 1.0,
774
+ "hidden_tests_passed": 1.0
775
+ },
776
+ "task_id": "vit_cifar10"
777
+ },
778
+ {
779
+ "primitive_type": "RemoveDeprecatedMethod",
780
+ "breakage_params": {
781
+ "class_name": "Trainer",
782
+ "method_name": "save_model",
783
+ "replacement": "save_to_hub"
784
+ },
785
+ "error_signature": "",
786
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -38,4 +38,4 @@\n trainer = Trainer(model=model, args=training_args, train_dataset=dataset)\n trainer.train()\n-trainer.save_model_DEPRECATED(\"/tmp/forge_output/checkpoint\")\n+trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
787
+ "visible_reward": 1.8,
788
+ "held_out": {
789
+ "executed_cleanly": 1.0,
790
+ "checkpoint_valid": 1.0,
791
+ "loss_decreased": 0.7984906001446131,
792
+ "metrics_in_range": 1.0,
793
+ "no_forbidden_workarounds": 1.0,
794
+ "intent_preserved": 1.0,
795
+ "hidden_tests_passed": 1.0
796
+ },
797
+ "task_id": "vit_cifar10"
798
+ },
799
+ {
800
+ "primitive_type": "RestructureDatasetSchema",
801
+ "breakage_params": {
802
+ "old_column": "text",
803
+ "new_column": "input_text"
804
+ },
805
+ "error_signature": "",
806
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -26,5 +26,5 @@\n answer = examples[\"answers\"][i]\n start_char = answer[\"answer_start\"][0]\n- end_char = start_char + len(answer[\"input_text\"][0])\n+ end_char = start_char + len(answer[\"text\"][0])\n \n token_start = next(\n",
807
+ "visible_reward": 1.8,
808
+ "held_out": {
809
+ "executed_cleanly": 1.0,
810
+ "checkpoint_valid": 1.0,
811
+ "loss_decreased": 0.7808289396602227,
812
+ "metrics_in_range": 1.0,
813
+ "no_forbidden_workarounds": 1.0,
814
+ "intent_preserved": 1.0,
815
+ "hidden_tests_passed": 1.0
816
+ },
817
+ "task_id": "albert_qa"
818
+ },
819
+ {
820
+ "primitive_type": "RestructureDatasetSchema",
821
+ "breakage_params": {
822
+ "old_column": "tokens",
823
+ "new_column": "words"
824
+ },
825
+ "error_signature": "",
826
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -14,5 +14,5 @@\n \n def tokenize_and_align(example):\n- enc = tokenizer(example[\"words\"], is_split_into_words=True, truncation=True, max_length=64)\n+ enc = tokenizer(example[\"tokens\"], is_split_into_words=True, truncation=True, max_length=64)\n word_ids = enc.word_ids()\n labels = []\n",
827
+ "visible_reward": 1.8,
828
+ "held_out": {
829
+ "executed_cleanly": 1.0,
830
+ "checkpoint_valid": 1.0,
831
+ "loss_decreased": 0.8699562543975037,
832
+ "metrics_in_range": 1.0,
833
+ "no_forbidden_workarounds": 1.0,
834
+ "intent_preserved": 1.0,
835
+ "hidden_tests_passed": 1.0
836
+ },
837
+ "task_id": "bert_ner"
838
+ },
839
+ {
840
+ "primitive_type": "RenameApiCall",
841
+ "breakage_params": {
842
+ "old_name": "trainer.train",
843
+ "new_name": "trainer.start_training"
844
+ },
845
+ "error_signature": "",
846
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -63,5 +63,5 @@\n data_collator=DefaultDataCollator(),\n )\n-trainer.start_training()\n+trainer.train()\n trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
847
+ "visible_reward": 1.8,
848
+ "held_out": {
849
+ "executed_cleanly": 1.0,
850
+ "checkpoint_valid": 1.0,
851
+ "loss_decreased": 0.911495927422025,
852
+ "metrics_in_range": 1.0,
853
+ "no_forbidden_workarounds": 1.0,
854
+ "intent_preserved": 1.0,
855
+ "hidden_tests_passed": 1.0
856
+ },
857
+ "task_id": "albert_qa"
858
+ },
859
+ {
860
+ "primitive_type": "RemoveDeprecatedMethod",
861
+ "breakage_params": {
862
+ "class_name": "Trainer",
863
+ "method_name": "save_model",
864
+ "replacement": "save_to_hub"
865
+ },
866
+ "error_signature": "",
867
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -64,4 +64,4 @@\n )\n trainer.train()\n-trainer.save_model_DEPRECATED(\"/tmp/forge_output/checkpoint\")\n+trainer.save_model(\"/tmp/forge_output/checkpoint\")\n print(\"TRAINING_COMPLETE\")\n",
868
+ "visible_reward": 1.8,
869
+ "held_out": {
870
+ "executed_cleanly": 1.0,
871
+ "checkpoint_valid": 1.0,
872
+ "loss_decreased": 0.6131321254553196,
873
+ "metrics_in_range": 1.0,
874
+ "no_forbidden_workarounds": 1.0,
875
+ "intent_preserved": 1.0,
876
+ "hidden_tests_passed": 1.0
877
+ },
878
+ "task_id": "albert_qa"
879
+ },
880
+ {
881
+ "primitive_type": "RestructureDatasetSchema",
882
+ "breakage_params": {
883
+ "old_column": "label",
884
+ "new_column": "labels"
885
+ },
886
+ "error_signature": "",
887
+ "repair_diff": "--- a/train.py\n+++ b/train.py\n@@ -22,5 +22,5 @@\n \n dataset = dataset.map(tokenize, batched=True)\n-dataset = dataset.rename_column(\"labels\", \"labels\")\n+dataset = dataset.rename_column(\"label\", \"labels\")\n dataset.set_format(\"torch\", columns=[\"input_ids\", \"attention_mask\", \"labels\"])\n \n",
888
+ "visible_reward": 1.8,
889
+ "held_out": {
890
+ "executed_cleanly": 1.0,
891
+ "checkpoint_valid": 1.0,
892
+ "loss_decreased": 0.6040748525323751,
893
+ "metrics_in_range": 1.0,
894
+ "no_forbidden_workarounds": 1.0,
895
+ "intent_preserved": 1.0,
896
+ "hidden_tests_passed": 1.0
897
+ },
898
+ "task_id": "electra_classification"
899
+ }
900
+ ],
901
+ "size": 43,
902
+ "by_primitive": {
903
+ "ChangeTokenizerBehavior": 7,
904
+ "RestructureDatasetSchema": 15,
905
+ "ChangeArgumentSignature": 7,
906
+ "RemoveDeprecatedMethod": 5,
907
+ "RenameApiCall": 6,
908
+ "ModifyConfigField": 3
909
+ }
910
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af5891a15588546db1ac7f2baf8fa94835a51a85c032c39793a55bb048b47446
3
+ size 11422523
tokenizer_config.json ADDED
@@ -0,0 +1,201 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "is_local": false,
9
+ "model_max_length": 32768,
10
+ "pad_token": "<|PAD_TOKEN|>",
11
+ "padding_side": "right",
12
+ "split_special_tokens": false,
13
+ "tokenizer_class": "Qwen2Tokenizer",
14
+ "unk_token": null,
15
+ "added_tokens_decoder": {
16
+ "151643": {
17
+ "content": "<|endoftext|>",
18
+ "single_word": false,
19
+ "lstrip": false,
20
+ "rstrip": false,
21
+ "normalized": false,
22
+ "special": true
23
+ },
24
+ "151644": {
25
+ "content": "<|im_start|>",
26
+ "single_word": false,
27
+ "lstrip": false,
28
+ "rstrip": false,
29
+ "normalized": false,
30
+ "special": true
31
+ },
32
+ "151645": {
33
+ "content": "<|im_end|>",
34
+ "single_word": false,
35
+ "lstrip": false,
36
+ "rstrip": false,
37
+ "normalized": false,
38
+ "special": true
39
+ },
40
+ "151646": {
41
+ "content": "<|object_ref_start|>",
42
+ "single_word": false,
43
+ "lstrip": false,
44
+ "rstrip": false,
45
+ "normalized": false,
46
+ "special": true
47
+ },
48
+ "151647": {
49
+ "content": "<|object_ref_end|>",
50
+ "single_word": false,
51
+ "lstrip": false,
52
+ "rstrip": false,
53
+ "normalized": false,
54
+ "special": true
55
+ },
56
+ "151648": {
57
+ "content": "<|box_start|>",
58
+ "single_word": false,
59
+ "lstrip": false,
60
+ "rstrip": false,
61
+ "normalized": false,
62
+ "special": true
63
+ },
64
+ "151649": {
65
+ "content": "<|box_end|>",
66
+ "single_word": false,
67
+ "lstrip": false,
68
+ "rstrip": false,
69
+ "normalized": false,
70
+ "special": true
71
+ },
72
+ "151650": {
73
+ "content": "<|quad_start|>",
74
+ "single_word": false,
75
+ "lstrip": false,
76
+ "rstrip": false,
77
+ "normalized": false,
78
+ "special": true
79
+ },
80
+ "151651": {
81
+ "content": "<|quad_end|>",
82
+ "single_word": false,
83
+ "lstrip": false,
84
+ "rstrip": false,
85
+ "normalized": false,
86
+ "special": true
87
+ },
88
+ "151652": {
89
+ "content": "<|vision_start|>",
90
+ "single_word": false,
91
+ "lstrip": false,
92
+ "rstrip": false,
93
+ "normalized": false,
94
+ "special": true
95
+ },
96
+ "151653": {
97
+ "content": "<|vision_end|>",
98
+ "single_word": false,
99
+ "lstrip": false,
100
+ "rstrip": false,
101
+ "normalized": false,
102
+ "special": true
103
+ },
104
+ "151654": {
105
+ "content": "<|vision_pad|>",
106
+ "single_word": false,
107
+ "lstrip": false,
108
+ "rstrip": false,
109
+ "normalized": false,
110
+ "special": true
111
+ },
112
+ "151655": {
113
+ "content": "<|image_pad|>",
114
+ "single_word": false,
115
+ "lstrip": false,
116
+ "rstrip": false,
117
+ "normalized": false,
118
+ "special": true
119
+ },
120
+ "151656": {
121
+ "content": "<|video_pad|>",
122
+ "single_word": false,
123
+ "lstrip": false,
124
+ "rstrip": false,
125
+ "normalized": false,
126
+ "special": true
127
+ },
128
+ "151657": {
129
+ "content": "<tool_call>",
130
+ "single_word": false,
131
+ "lstrip": false,
132
+ "rstrip": false,
133
+ "normalized": false,
134
+ "special": false
135
+ },
136
+ "151658": {
137
+ "content": "</tool_call>",
138
+ "single_word": false,
139
+ "lstrip": false,
140
+ "rstrip": false,
141
+ "normalized": false,
142
+ "special": false
143
+ },
144
+ "151659": {
145
+ "content": "<|fim_prefix|>",
146
+ "single_word": false,
147
+ "lstrip": false,
148
+ "rstrip": false,
149
+ "normalized": false,
150
+ "special": false
151
+ },
152
+ "151660": {
153
+ "content": "<|fim_middle|>",
154
+ "single_word": false,
155
+ "lstrip": false,
156
+ "rstrip": false,
157
+ "normalized": false,
158
+ "special": false
159
+ },
160
+ "151661": {
161
+ "content": "<|fim_suffix|>",
162
+ "single_word": false,
163
+ "lstrip": false,
164
+ "rstrip": false,
165
+ "normalized": false,
166
+ "special": false
167
+ },
168
+ "151662": {
169
+ "content": "<|fim_pad|>",
170
+ "single_word": false,
171
+ "lstrip": false,
172
+ "rstrip": false,
173
+ "normalized": false,
174
+ "special": false
175
+ },
176
+ "151663": {
177
+ "content": "<|repo_name|>",
178
+ "single_word": false,
179
+ "lstrip": false,
180
+ "rstrip": false,
181
+ "normalized": false,
182
+ "special": false
183
+ },
184
+ "151664": {
185
+ "content": "<|file_sep|>",
186
+ "single_word": false,
187
+ "lstrip": false,
188
+ "rstrip": false,
189
+ "normalized": false,
190
+ "special": false
191
+ },
192
+ "151665": {
193
+ "content": "<|PAD_TOKEN|>",
194
+ "single_word": false,
195
+ "lstrip": false,
196
+ "rstrip": false,
197
+ "normalized": false,
198
+ "special": true
199
+ }
200
+ }
201
+ }