B4K2xx commited on
Commit
f296a89
·
verified ·
1 Parent(s): 42f3426

Add m22 weights and results

Browse files
m22/benchmark_results.json ADDED
@@ -0,0 +1,540 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "m22",
3
+ "checkpoint": "results/m22/final.pt",
4
+ "trained_step": 15258,
5
+ "num_fewshot": 0,
6
+ "limit": null,
7
+ "results": {
8
+ "lambada_openai": {
9
+ "name": "lambada_openai",
10
+ "alias": "lambada_openai",
11
+ "sample_len": 5153,
12
+ "perplexity,none": 785.2879050413346,
13
+ "perplexity_stderr,none": 43.50157604743723,
14
+ "acc,none": 0.17019212109450804,
15
+ "acc_stderr,none": 0.005235649623696734
16
+ },
17
+ "hellaswag": {
18
+ "name": "hellaswag",
19
+ "alias": "hellaswag",
20
+ "sample_len": 10042,
21
+ "acc,none": 0.27145986855208126,
22
+ "acc_stderr,none": 0.004438038583345241,
23
+ "acc_norm,none": 0.2756423023302131,
24
+ "acc_norm_stderr,none": 0.004459241474518529
25
+ },
26
+ "piqa": {
27
+ "name": "piqa",
28
+ "alias": "piqa",
29
+ "sample_len": 1838,
30
+ "acc,none": 0.5946681175190425,
31
+ "acc_stderr,none": 0.011454816387346781,
32
+ "acc_norm,none": 0.5778019586507073,
33
+ "acc_norm_stderr,none": 0.01152372892677764
34
+ },
35
+ "sciq": {
36
+ "name": "sciq",
37
+ "alias": "sciq",
38
+ "sample_len": 1000,
39
+ "acc,none": 0.691,
40
+ "acc_stderr,none": 0.014619600977206347,
41
+ "acc_norm,none": 0.619,
42
+ "acc_norm_stderr,none": 0.015364734787007391
43
+ },
44
+ "arc_easy": {
45
+ "name": "arc_easy",
46
+ "alias": "arc_easy",
47
+ "sample_len": 2376,
48
+ "acc,none": 0.43392255892255893,
49
+ "acc_stderr,none": 0.010169795770462075,
50
+ "acc_norm,none": 0.38762626262626265,
51
+ "acc_norm_stderr,none": 0.009997307914447627
52
+ },
53
+ "winogrande": {
54
+ "name": "winogrande",
55
+ "alias": "winogrande",
56
+ "sample_len": 1267,
57
+ "acc,none": 0.5169692186266772,
58
+ "acc_stderr,none": 0.014044390401613116
59
+ },
60
+ "blimp_adjunct_island": {
61
+ "name": "blimp_adjunct_island",
62
+ "alias": "blimp_adjunct_island",
63
+ "sample_len": 1000,
64
+ "acc,none": 0.796,
65
+ "acc_stderr,none": 0.012749374359024311
66
+ },
67
+ "blimp_anaphor_gender_agreement": {
68
+ "name": "blimp_anaphor_gender_agreement",
69
+ "alias": "blimp_anaphor_gender_agreement",
70
+ "sample_len": 1000,
71
+ "acc,none": 0.911,
72
+ "acc_stderr,none": 0.009008893392651516
73
+ },
74
+ "blimp_anaphor_number_agreement": {
75
+ "name": "blimp_anaphor_number_agreement",
76
+ "alias": "blimp_anaphor_number_agreement",
77
+ "sample_len": 1000,
78
+ "acc,none": 0.975,
79
+ "acc_stderr,none": 0.004939574819698493
80
+ },
81
+ "blimp_animate_subject_passive": {
82
+ "name": "blimp_animate_subject_passive",
83
+ "alias": "blimp_animate_subject_passive",
84
+ "sample_len": 1000,
85
+ "acc,none": 0.739,
86
+ "acc_stderr,none": 0.013895037677965183
87
+ },
88
+ "blimp_animate_subject_trans": {
89
+ "name": "blimp_animate_subject_trans",
90
+ "alias": "blimp_animate_subject_trans",
91
+ "sample_len": 1000,
92
+ "acc,none": 0.894,
93
+ "acc_stderr,none": 0.00973955126578524
94
+ },
95
+ "blimp_causative": {
96
+ "name": "blimp_causative",
97
+ "alias": "blimp_causative",
98
+ "sample_len": 1000,
99
+ "acc,none": 0.718,
100
+ "acc_stderr,none": 0.01423652621529138
101
+ },
102
+ "blimp_complex_NP_island": {
103
+ "name": "blimp_complex_NP_island",
104
+ "alias": "blimp_complex_NP_island",
105
+ "sample_len": 1000,
106
+ "acc,none": 0.481,
107
+ "acc_stderr,none": 0.015807874268505835
108
+ },
109
+ "blimp_coordinate_structure_constraint_complex_left_branch": {
110
+ "name": "blimp_coordinate_structure_constraint_complex_left_branch",
111
+ "alias": "blimp_coordinate_structure_constraint_complex_left_branch",
112
+ "sample_len": 1000,
113
+ "acc,none": 0.415,
114
+ "acc_stderr,none": 0.015589035185604594
115
+ },
116
+ "blimp_coordinate_structure_constraint_object_extraction": {
117
+ "name": "blimp_coordinate_structure_constraint_object_extraction",
118
+ "alias": "blimp_coordinate_structure_constraint_object_extraction",
119
+ "sample_len": 1000,
120
+ "acc,none": 0.801,
121
+ "acc_stderr,none": 0.01263164908309911
122
+ },
123
+ "blimp_determiner_noun_agreement_1": {
124
+ "name": "blimp_determiner_noun_agreement_1",
125
+ "alias": "blimp_determiner_noun_agreement_1",
126
+ "sample_len": 1000,
127
+ "acc,none": 0.972,
128
+ "acc_stderr,none": 0.005219506034410081
129
+ },
130
+ "blimp_determiner_noun_agreement_2": {
131
+ "name": "blimp_determiner_noun_agreement_2",
132
+ "alias": "blimp_determiner_noun_agreement_2",
133
+ "sample_len": 1000,
134
+ "acc,none": 0.942,
135
+ "acc_stderr,none": 0.0073953154557929906
136
+ },
137
+ "blimp_determiner_noun_agreement_irregular_1": {
138
+ "name": "blimp_determiner_noun_agreement_irregular_1",
139
+ "alias": "blimp_determiner_noun_agreement_irregular_1",
140
+ "sample_len": 1000,
141
+ "acc,none": 0.88,
142
+ "acc_stderr,none": 0.010281328012747462
143
+ },
144
+ "blimp_determiner_noun_agreement_irregular_2": {
145
+ "name": "blimp_determiner_noun_agreement_irregular_2",
146
+ "alias": "blimp_determiner_noun_agreement_irregular_2",
147
+ "sample_len": 1000,
148
+ "acc,none": 0.924,
149
+ "acc_stderr,none": 0.008384169266796462
150
+ },
151
+ "blimp_determiner_noun_agreement_with_adj_2": {
152
+ "name": "blimp_determiner_noun_agreement_with_adj_2",
153
+ "alias": "blimp_determiner_noun_agreement_with_adj_2",
154
+ "sample_len": 1000,
155
+ "acc,none": 0.889,
156
+ "acc_stderr,none": 0.009938701010583716
157
+ },
158
+ "blimp_determiner_noun_agreement_with_adj_irregular_1": {
159
+ "name": "blimp_determiner_noun_agreement_with_adj_irregular_1",
160
+ "alias": "blimp_determiner_noun_agreement_with_adj_irregular_1",
161
+ "sample_len": 1000,
162
+ "acc,none": 0.827,
163
+ "acc_stderr,none": 0.011967214137559927
164
+ },
165
+ "blimp_determiner_noun_agreement_with_adj_irregular_2": {
166
+ "name": "blimp_determiner_noun_agreement_with_adj_irregular_2",
167
+ "alias": "blimp_determiner_noun_agreement_with_adj_irregular_2",
168
+ "sample_len": 1000,
169
+ "acc,none": 0.87,
170
+ "acc_stderr,none": 0.010640169792499236
171
+ },
172
+ "blimp_determiner_noun_agreement_with_adjective_1": {
173
+ "name": "blimp_determiner_noun_agreement_with_adjective_1",
174
+ "alias": "blimp_determiner_noun_agreement_with_adjective_1",
175
+ "sample_len": 1000,
176
+ "acc,none": 0.916,
177
+ "acc_stderr,none": 0.008776162089491113
178
+ },
179
+ "blimp_distractor_agreement_relational_noun": {
180
+ "name": "blimp_distractor_agreement_relational_noun",
181
+ "alias": "blimp_distractor_agreement_relational_noun",
182
+ "sample_len": 1000,
183
+ "acc,none": 0.853,
184
+ "acc_stderr,none": 0.011203415395160376
185
+ },
186
+ "blimp_distractor_agreement_relative_clause": {
187
+ "name": "blimp_distractor_agreement_relative_clause",
188
+ "alias": "blimp_distractor_agreement_relative_clause",
189
+ "sample_len": 1000,
190
+ "acc,none": 0.684,
191
+ "acc_stderr,none": 0.014709193056057166
192
+ },
193
+ "blimp_drop_argument": {
194
+ "name": "blimp_drop_argument",
195
+ "alias": "blimp_drop_argument",
196
+ "sample_len": 1000,
197
+ "acc,none": 0.713,
198
+ "acc_stderr,none": 0.014312087053810001
199
+ },
200
+ "blimp_ellipsis_n_bar_1": {
201
+ "name": "blimp_ellipsis_n_bar_1",
202
+ "alias": "blimp_ellipsis_n_bar_1",
203
+ "sample_len": 1000,
204
+ "acc,none": 0.689,
205
+ "acc_stderr,none": 0.01464559638572269
206
+ },
207
+ "blimp_ellipsis_n_bar_2": {
208
+ "name": "blimp_ellipsis_n_bar_2",
209
+ "alias": "blimp_ellipsis_n_bar_2",
210
+ "sample_len": 1000,
211
+ "acc,none": 0.866,
212
+ "acc_stderr,none": 0.010777762298369738
213
+ },
214
+ "blimp_existential_there_object_raising": {
215
+ "name": "blimp_existential_there_object_raising",
216
+ "alias": "blimp_existential_there_object_raising",
217
+ "sample_len": 1000,
218
+ "acc,none": 0.696,
219
+ "acc_stderr,none": 0.014553205687950535
220
+ },
221
+ "blimp_existential_there_quantifiers_1": {
222
+ "name": "blimp_existential_there_quantifiers_1",
223
+ "alias": "blimp_existential_there_quantifiers_1",
224
+ "sample_len": 1000,
225
+ "acc,none": 0.972,
226
+ "acc_stderr,none": 0.005219506034410081
227
+ },
228
+ "blimp_existential_there_quantifiers_2": {
229
+ "name": "blimp_existential_there_quantifiers_2",
230
+ "alias": "blimp_existential_there_quantifiers_2",
231
+ "sample_len": 1000,
232
+ "acc,none": 0.368,
233
+ "acc_stderr,none": 0.015258073561521743
234
+ },
235
+ "blimp_existential_there_subject_raising": {
236
+ "name": "blimp_existential_there_subject_raising",
237
+ "alias": "blimp_existential_there_subject_raising",
238
+ "sample_len": 1000,
239
+ "acc,none": 0.834,
240
+ "acc_stderr,none": 0.011772110370812133
241
+ },
242
+ "blimp_expletive_it_object_raising": {
243
+ "name": "blimp_expletive_it_object_raising",
244
+ "alias": "blimp_expletive_it_object_raising",
245
+ "sample_len": 1000,
246
+ "acc,none": 0.732,
247
+ "acc_stderr,none": 0.014013292702729519
248
+ },
249
+ "blimp_inchoative": {
250
+ "name": "blimp_inchoative",
251
+ "alias": "blimp_inchoative",
252
+ "sample_len": 1000,
253
+ "acc,none": 0.611,
254
+ "acc_stderr,none": 0.015424555647308514
255
+ },
256
+ "blimp_intransitive": {
257
+ "name": "blimp_intransitive",
258
+ "alias": "blimp_intransitive",
259
+ "sample_len": 1000,
260
+ "acc,none": 0.72,
261
+ "acc_stderr,none": 0.014205696104091548
262
+ },
263
+ "blimp_irregular_past_participle_adjectives": {
264
+ "name": "blimp_irregular_past_participle_adjectives",
265
+ "alias": "blimp_irregular_past_participle_adjectives",
266
+ "sample_len": 1000,
267
+ "acc,none": 0.816,
268
+ "acc_stderr,none": 0.01225945734093864
269
+ },
270
+ "blimp_irregular_past_participle_verbs": {
271
+ "name": "blimp_irregular_past_participle_verbs",
272
+ "alias": "blimp_irregular_past_participle_verbs",
273
+ "sample_len": 1000,
274
+ "acc,none": 0.78,
275
+ "acc_stderr,none": 0.013106173040661862
276
+ },
277
+ "blimp_irregular_plural_subject_verb_agreement_1": {
278
+ "name": "blimp_irregular_plural_subject_verb_agreement_1",
279
+ "alias": "blimp_irregular_plural_subject_verb_agreement_1",
280
+ "sample_len": 1000,
281
+ "acc,none": 0.857,
282
+ "acc_stderr,none": 0.011075814808567074
283
+ },
284
+ "blimp_irregular_plural_subject_verb_agreement_2": {
285
+ "name": "blimp_irregular_plural_subject_verb_agreement_2",
286
+ "alias": "blimp_irregular_plural_subject_verb_agreement_2",
287
+ "sample_len": 1000,
288
+ "acc,none": 0.87,
289
+ "acc_stderr,none": 0.010640169792499236
290
+ },
291
+ "blimp_left_branch_island_echo_question": {
292
+ "name": "blimp_left_branch_island_echo_question",
293
+ "alias": "blimp_left_branch_island_echo_question",
294
+ "sample_len": 1000,
295
+ "acc,none": 0.536,
296
+ "acc_stderr,none": 0.015778243024904673
297
+ },
298
+ "blimp_left_branch_island_simple_question": {
299
+ "name": "blimp_left_branch_island_simple_question",
300
+ "alias": "blimp_left_branch_island_simple_question",
301
+ "sample_len": 1000,
302
+ "acc,none": 0.514,
303
+ "acc_stderr,none": 0.01581309754773093
304
+ },
305
+ "blimp_matrix_question_npi_licensor_present": {
306
+ "name": "blimp_matrix_question_npi_licensor_present",
307
+ "alias": "blimp_matrix_question_npi_licensor_present",
308
+ "sample_len": 1000,
309
+ "acc,none": 0.414,
310
+ "acc_stderr,none": 0.015583544104177552
311
+ },
312
+ "blimp_npi_present_1": {
313
+ "name": "blimp_npi_present_1",
314
+ "alias": "blimp_npi_present_1",
315
+ "sample_len": 1000,
316
+ "acc,none": 0.573,
317
+ "acc_stderr,none": 0.015649789644462186
318
+ },
319
+ "blimp_npi_present_2": {
320
+ "name": "blimp_npi_present_2",
321
+ "alias": "blimp_npi_present_2",
322
+ "sample_len": 1000,
323
+ "acc,none": 0.563,
324
+ "acc_stderr,none": 0.01569322392873043
325
+ },
326
+ "blimp_only_npi_licensor_present": {
327
+ "name": "blimp_only_npi_licensor_present",
328
+ "alias": "blimp_only_npi_licensor_present",
329
+ "sample_len": 1000,
330
+ "acc,none": 0.738,
331
+ "acc_stderr,none": 0.013912208651021217
332
+ },
333
+ "blimp_only_npi_scope": {
334
+ "name": "blimp_only_npi_scope",
335
+ "alias": "blimp_only_npi_scope",
336
+ "sample_len": 1000,
337
+ "acc,none": 0.486,
338
+ "acc_stderr,none": 0.01581309754773093
339
+ },
340
+ "blimp_passive_1": {
341
+ "name": "blimp_passive_1",
342
+ "alias": "blimp_passive_1",
343
+ "sample_len": 1000,
344
+ "acc,none": 0.88,
345
+ "acc_stderr,none": 0.010281328012747462
346
+ },
347
+ "blimp_passive_2": {
348
+ "name": "blimp_passive_2",
349
+ "alias": "blimp_passive_2",
350
+ "sample_len": 1000,
351
+ "acc,none": 0.865,
352
+ "acc_stderr,none": 0.010811655372416006
353
+ },
354
+ "blimp_principle_A_c_command": {
355
+ "name": "blimp_principle_A_c_command",
356
+ "alias": "blimp_principle_A_c_command",
357
+ "sample_len": 1000,
358
+ "acc,none": 0.623,
359
+ "acc_stderr,none": 0.0153331701257798
360
+ },
361
+ "blimp_principle_A_case_1": {
362
+ "name": "blimp_principle_A_case_1",
363
+ "alias": "blimp_principle_A_case_1",
364
+ "sample_len": 1000,
365
+ "acc,none": 1.0,
366
+ "acc_stderr,none": 0.0
367
+ },
368
+ "blimp_principle_A_case_2": {
369
+ "name": "blimp_principle_A_case_2",
370
+ "alias": "blimp_principle_A_case_2",
371
+ "sample_len": 1000,
372
+ "acc,none": 0.943,
373
+ "acc_stderr,none": 0.007335175853706905
374
+ },
375
+ "blimp_principle_A_domain_1": {
376
+ "name": "blimp_principle_A_domain_1",
377
+ "alias": "blimp_principle_A_domain_1",
378
+ "sample_len": 1000,
379
+ "acc,none": 0.9,
380
+ "acc_stderr,none": 0.00949157995752499
381
+ },
382
+ "blimp_principle_A_domain_2": {
383
+ "name": "blimp_principle_A_domain_2",
384
+ "alias": "blimp_principle_A_domain_2",
385
+ "sample_len": 1000,
386
+ "acc,none": 0.745,
387
+ "acc_stderr,none": 0.013790038620872863
388
+ },
389
+ "blimp_principle_A_domain_3": {
390
+ "name": "blimp_principle_A_domain_3",
391
+ "alias": "blimp_principle_A_domain_3",
392
+ "sample_len": 1000,
393
+ "acc,none": 0.613,
394
+ "acc_stderr,none": 0.015410011955494015
395
+ },
396
+ "blimp_principle_A_reconstruction": {
397
+ "name": "blimp_principle_A_reconstruction",
398
+ "alias": "blimp_principle_A_reconstruction",
399
+ "sample_len": 1000,
400
+ "acc,none": 0.488,
401
+ "acc_stderr,none": 0.015814743314581686
402
+ },
403
+ "blimp_regular_plural_subject_verb_agreement_1": {
404
+ "name": "blimp_regular_plural_subject_verb_agreement_1",
405
+ "alias": "blimp_regular_plural_subject_verb_agreement_1",
406
+ "sample_len": 1000,
407
+ "acc,none": 0.834,
408
+ "acc_stderr,none": 0.011772110370812133
409
+ },
410
+ "blimp_regular_plural_subject_verb_agreement_2": {
411
+ "name": "blimp_regular_plural_subject_verb_agreement_2",
412
+ "alias": "blimp_regular_plural_subject_verb_agreement_2",
413
+ "sample_len": 1000,
414
+ "acc,none": 0.881,
415
+ "acc_stderr,none": 0.01024421514533661
416
+ },
417
+ "blimp_sentential_negation_npi_licensor_present": {
418
+ "name": "blimp_sentential_negation_npi_licensor_present",
419
+ "alias": "blimp_sentential_negation_npi_licensor_present",
420
+ "sample_len": 1000,
421
+ "acc,none": 0.98,
422
+ "acc_stderr,none": 0.004429403980178329
423
+ },
424
+ "blimp_sentential_negation_npi_scope": {
425
+ "name": "blimp_sentential_negation_npi_scope",
426
+ "alias": "blimp_sentential_negation_npi_scope",
427
+ "sample_len": 1000,
428
+ "acc,none": 0.388,
429
+ "acc_stderr,none": 0.015417317979911216
430
+ },
431
+ "blimp_sentential_subject_island": {
432
+ "name": "blimp_sentential_subject_island",
433
+ "alias": "blimp_sentential_subject_island",
434
+ "sample_len": 1000,
435
+ "acc,none": 0.546,
436
+ "acc_stderr,none": 0.01575221038877186
437
+ },
438
+ "blimp_superlative_quantifiers_1": {
439
+ "name": "blimp_superlative_quantifiers_1",
440
+ "alias": "blimp_superlative_quantifiers_1",
441
+ "sample_len": 1000,
442
+ "acc,none": 0.82,
443
+ "acc_stderr,none": 0.012155153135512022
444
+ },
445
+ "blimp_superlative_quantifiers_2": {
446
+ "name": "blimp_superlative_quantifiers_2",
447
+ "alias": "blimp_superlative_quantifiers_2",
448
+ "sample_len": 1000,
449
+ "acc,none": 0.797,
450
+ "acc_stderr,none": 0.01272607374459825
451
+ },
452
+ "blimp_tough_vs_raising_1": {
453
+ "name": "blimp_tough_vs_raising_1",
454
+ "alias": "blimp_tough_vs_raising_1",
455
+ "sample_len": 1000,
456
+ "acc,none": 0.523,
457
+ "acc_stderr,none": 0.015802554246726087
458
+ },
459
+ "blimp_tough_vs_raising_2": {
460
+ "name": "blimp_tough_vs_raising_2",
461
+ "alias": "blimp_tough_vs_raising_2",
462
+ "sample_len": 1000,
463
+ "acc,none": 0.881,
464
+ "acc_stderr,none": 0.01024421514533661
465
+ },
466
+ "blimp_transitive": {
467
+ "name": "blimp_transitive",
468
+ "alias": "blimp_transitive",
469
+ "sample_len": 1000,
470
+ "acc,none": 0.822,
471
+ "acc_stderr,none": 0.012102167676183571
472
+ },
473
+ "blimp_wh_island": {
474
+ "name": "blimp_wh_island",
475
+ "alias": "blimp_wh_island",
476
+ "sample_len": 1000,
477
+ "acc,none": 0.611,
478
+ "acc_stderr,none": 0.015424555647308514
479
+ },
480
+ "blimp_wh_questions_object_gap": {
481
+ "name": "blimp_wh_questions_object_gap",
482
+ "alias": "blimp_wh_questions_object_gap",
483
+ "sample_len": 1000,
484
+ "acc,none": 0.734,
485
+ "acc_stderr,none": 0.013979965645145179
486
+ },
487
+ "blimp_wh_questions_subject_gap": {
488
+ "name": "blimp_wh_questions_subject_gap",
489
+ "alias": "blimp_wh_questions_subject_gap",
490
+ "sample_len": 1000,
491
+ "acc,none": 0.919,
492
+ "acc_stderr,none": 0.008632121032139986
493
+ },
494
+ "blimp_wh_questions_subject_gap_long_distance": {
495
+ "name": "blimp_wh_questions_subject_gap_long_distance",
496
+ "alias": "blimp_wh_questions_subject_gap_long_distance",
497
+ "sample_len": 1000,
498
+ "acc,none": 0.889,
499
+ "acc_stderr,none": 0.009938701010583716
500
+ },
501
+ "blimp_wh_vs_that_no_gap": {
502
+ "name": "blimp_wh_vs_that_no_gap",
503
+ "alias": "blimp_wh_vs_that_no_gap",
504
+ "sample_len": 1000,
505
+ "acc,none": 0.968,
506
+ "acc_stderr,none": 0.005568393575081328
507
+ },
508
+ "blimp_wh_vs_that_no_gap_long_distance": {
509
+ "name": "blimp_wh_vs_that_no_gap_long_distance",
510
+ "alias": "blimp_wh_vs_that_no_gap_long_distance",
511
+ "sample_len": 1000,
512
+ "acc,none": 0.986,
513
+ "acc_stderr,none": 0.003717232548256541
514
+ },
515
+ "blimp_wh_vs_that_with_gap": {
516
+ "name": "blimp_wh_vs_that_with_gap",
517
+ "alias": "blimp_wh_vs_that_with_gap",
518
+ "sample_len": 1000,
519
+ "acc,none": 0.52,
520
+ "acc_stderr,none": 0.015806639423035177
521
+ },
522
+ "blimp_wh_vs_that_with_gap_long_distance": {
523
+ "name": "blimp_wh_vs_that_with_gap_long_distance",
524
+ "alias": "blimp_wh_vs_that_with_gap_long_distance",
525
+ "sample_len": 1000,
526
+ "acc,none": 0.213,
527
+ "acc_stderr,none": 0.012953717566737312
528
+ },
529
+ "blimp": {
530
+ "alias": "blimp",
531
+ "name": "blimp",
532
+ "sample_len": 67000,
533
+ "acc,none": 0.7493134328358209,
534
+ "acc_stderr,none": 0.0015197586373841688,
535
+ "sample_count": {
536
+ "acc,none": 67000
537
+ }
538
+ }
539
+ }
540
+ }
m22/config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "m22",
3
+ "run_name": "m22-mha-learned-dense-adamw-seed1338",
4
+ "data_dir": "data/fineweb-edu",
5
+ "vocab_size": 50257,
6
+ "block_size": 1024,
7
+ "d_model": 512,
8
+ "num_layers": 12,
9
+ "num_heads": 8,
10
+ "ffn_mult": 4,
11
+ "dropout": 0.0,
12
+ "bias": false,
13
+ "norm_type": "rmsnorm",
14
+ "tie_embeddings": true,
15
+ "attention_type": "mha",
16
+ "gqa_num_kv_heads": 2,
17
+ "mla_latent_dim": 64,
18
+ "pos_type": "learned",
19
+ "rope_base": 10000,
20
+ "ffn_type": "dense",
21
+ "moe_num_experts": 8,
22
+ "moe_top_k": 2,
23
+ "optimizer": "adamw",
24
+ "lr": 0.0006,
25
+ "min_lr": 6e-05,
26
+ "weight_decay": 0.1,
27
+ "beta1": 0.9,
28
+ "beta2": 0.95,
29
+ "grad_clip": 1.0,
30
+ "muon_lr": 0.02,
31
+ "muon_momentum": 0.95,
32
+ "training_tokens": 1000000000,
33
+ "warmup_tokens": 6000000,
34
+ "lr_schedule": "cosine",
35
+ "batch_size": 4,
36
+ "grad_accum_steps": 16,
37
+ "device": "cuda",
38
+ "dtype": "bfloat16",
39
+ "compile": true,
40
+ "seed": 1338,
41
+ "eval_interval": 250,
42
+ "eval_iters": 100,
43
+ "log_interval": 10,
44
+ "checkpoint_interval": 1000,
45
+ "results_dir": "results",
46
+ "mlflow_experiment": "slm-ablation",
47
+ "mlflow_tracking_uri": "sqlite:///mlflow.db",
48
+ "mlflow_artifact_location": "./mlartifacts",
49
+ "seed_group": "m01"
50
+ }
m22/eval_results.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "m22",
3
+ "checkpoint": "/home/akshat/codemindv2/results/m22/final.pt",
4
+ "trained_step": 15258,
5
+ "total_params": 64017408,
6
+ "active_params": 64017408,
7
+ "flops_per_forward": 128034816.0,
8
+ "val_loss": 3.5585316568613052,
9
+ "val_perplexity": 35.11160339377922,
10
+ "bits_per_token": 5.133875974200191,
11
+ "ppl_tokens_evaluated": 655360,
12
+ "prefill_tok_per_sec": 195315.3982182166,
13
+ "inference_tok_per_sec": 326.0861900573855,
14
+ "inference_tok_per_sec_kvcache": null,
15
+ "kvcache_speedup": null,
16
+ "gen_tokens": 128,
17
+ "peak_infer_vram_gb": 0.394,
18
+ "peak_train_vram_gb": 4.093,
19
+ "closeness_answer_ppl": 33.765,
20
+ "closeness_hit_rate": 0.45,
21
+ "closeness_token_f1": 0.093,
22
+ "closeness_n": 20,
23
+ "closeness_items": [
24
+ {
25
+ "prompt": "The capital of France is",
26
+ "answer": " Paris",
27
+ "generated": " the capital of France.\n",
28
+ "answer_nll": 3.2356,
29
+ "answer_ppl": 25.421,
30
+ "hit": false,
31
+ "token_f1": 0.0
32
+ },
33
+ {
34
+ "prompt": "The capital of Japan is",
35
+ "answer": " Tokyo",
36
+ "generated": " the capital of the United States",
37
+ "answer_nll": 2.6915,
38
+ "answer_ppl": 14.754,
39
+ "hit": false,
40
+ "token_f1": 0.0
41
+ },
42
+ {
43
+ "prompt": "The largest planet in the solar system is",
44
+ "answer": " Jupiter",
45
+ "generated": " the Sun, which is the",
46
+ "answer_nll": 3.1711,
47
+ "answer_ppl": 23.834,
48
+ "hit": false,
49
+ "token_f1": 0.0
50
+ },
51
+ {
52
+ "prompt": "Water is made of hydrogen and",
53
+ "answer": " oxygen",
54
+ "generated": " oxygen.\n- The oxygen",
55
+ "answer_nll": 0.9604,
56
+ "answer_ppl": 2.613,
57
+ "hit": true,
58
+ "token_f1": 0.4
59
+ },
60
+ {
61
+ "prompt": "The opposite of hot is",
62
+ "answer": " cold",
63
+ "generated": " the coldest, the cold",
64
+ "answer_nll": 3.1722,
65
+ "answer_ppl": 23.86,
66
+ "hit": true,
67
+ "token_f1": 0.4
68
+ },
69
+ {
70
+ "prompt": "The opposite of up is",
71
+ "answer": " down",
72
+ "generated": " the fact that the two are",
73
+ "answer_nll": 6.9194,
74
+ "answer_ppl": 1011.664,
75
+ "hit": false,
76
+ "token_f1": 0.0
77
+ },
78
+ {
79
+ "prompt": "Two plus two equals",
80
+ "answer": " four",
81
+ "generated": ", and the two equals,",
82
+ "answer_nll": 5.168,
83
+ "answer_ppl": 175.56,
84
+ "hit": false,
85
+ "token_f1": 0.0
86
+ },
87
+ {
88
+ "prompt": "The sun rises in the",
89
+ "answer": " east",
90
+ "generated": " sky, and the sun rises",
91
+ "answer_nll": 3.1618,
92
+ "answer_ppl": 23.613,
93
+ "hit": false,
94
+ "token_f1": 0.0
95
+ },
96
+ {
97
+ "prompt": "The first president of the United States was George",
98
+ "answer": " Washington",
99
+ "generated": " Washington, who was the first",
100
+ "answer_nll": 0.9015,
101
+ "answer_ppl": 2.463,
102
+ "hit": true,
103
+ "token_f1": 0.0
104
+ },
105
+ {
106
+ "prompt": "The speed of light is approximately 300,000 kilometers per",
107
+ "answer": " second",
108
+ "generated": " second.\nThe speed of",
109
+ "answer_nll": 0.2886,
110
+ "answer_ppl": 1.335,
111
+ "hit": true,
112
+ "token_f1": 0.0
113
+ },
114
+ {
115
+ "prompt": "A triangle has three",
116
+ "answer": " sides",
117
+ "generated": " sides, and the sides are",
118
+ "answer_nll": 2.1492,
119
+ "answer_ppl": 8.578,
120
+ "hit": true,
121
+ "token_f1": 0.333
122
+ },
123
+ {
124
+ "prompt": "The chemical symbol for gold is",
125
+ "answer": " Au",
126
+ "generated": " the gold of the world.",
127
+ "answer_nll": 9.4926,
128
+ "answer_ppl": 13261.109,
129
+ "hit": false,
130
+ "token_f1": 0.0
131
+ },
132
+ {
133
+ "prompt": "The freezing point of water in Celsius is",
134
+ "answer": " zero",
135
+ "generated": " about 1.5 degrees Celsius",
136
+ "answer_nll": 5.8433,
137
+ "answer_ppl": 344.908,
138
+ "hit": false,
139
+ "token_f1": 0.0
140
+ },
141
+ {
142
+ "prompt": "The author of Romeo and Juliet is William",
143
+ "answer": " Shakespeare",
144
+ "generated": " Shakespeare.\nThe author of",
145
+ "answer_nll": 1.3143,
146
+ "answer_ppl": 3.722,
147
+ "hit": true,
148
+ "token_f1": 0.0
149
+ },
150
+ {
151
+ "prompt": "The Earth revolves around the",
152
+ "answer": " Sun",
153
+ "generated": " Sun, and the Sun revolves",
154
+ "answer_nll": 0.9692,
155
+ "answer_ppl": 2.636,
156
+ "hit": true,
157
+ "token_f1": 0.333
158
+ },
159
+ {
160
+ "prompt": "The plural of mouse is",
161
+ "answer": " mice",
162
+ "generated": " the plural of mouse.\n",
163
+ "answer_nll": 10.0879,
164
+ "answer_ppl": 24050.054,
165
+ "hit": false,
166
+ "token_f1": 0.0
167
+ },
168
+ {
169
+ "prompt": "Photosynthesis occurs in the",
170
+ "answer": " leaves",
171
+ "generated": " chlorophyll-rich waters",
172
+ "answer_nll": 3.7037,
173
+ "answer_ppl": 40.596,
174
+ "hit": false,
175
+ "token_f1": 0.0
176
+ },
177
+ {
178
+ "prompt": "The human heart pumps",
179
+ "answer": " blood",
180
+ "generated": " blood to the heart.\n",
181
+ "answer_nll": 1.0486,
182
+ "answer_ppl": 2.854,
183
+ "hit": true,
184
+ "token_f1": 0.4
185
+ },
186
+ {
187
+ "prompt": "The currency used in the United States is the",
188
+ "answer": " dollar",
189
+ "generated": " United States Dollar.\nThe",
190
+ "answer_nll": 3.1815,
191
+ "answer_ppl": 24.084,
192
+ "hit": true,
193
+ "token_f1": 0.0
194
+ },
195
+ {
196
+ "prompt": "The tallest animal in the world is the",
197
+ "answer": " giraffe",
198
+ "generated": " giant elephant.\nThe elephant",
199
+ "answer_nll": 2.928,
200
+ "answer_ppl": 18.69,
201
+ "hit": false,
202
+ "token_f1": 0.0
203
+ }
204
+ ],
205
+ "generation_samples": {
206
+ "The most important thing about sleep is": "The most important thing about sleep is waking up. It can be easy to go back to sleep for a while. So, it is important to give proper care to your sleep.\nIn some cases sleep is disrupted because of the lack of a proper sleep environment. If the body doesn\u2019t get enough sleep, your body can",
207
+ "Scientists have discovered that": "Scientists have discovered that when their muscles contract, it can trigger a reaction. This is how the muscles contract when they're stressed.\nTrying to work on the muscles in these situations is the best way to help your heart and brain function.\nYour body is only a little bit smaller than any other organs.\n",
208
+ "In the history of mathematics,": "In the history of mathematics, we had very much in this field of science a year before, we have only two problems which have always proved to us quite difficult.\nI shall attempt to describe our theory of an elementary school which we have only to discuss in terms of mathematics, and we shall say that the mathematics that we will",
209
+ "Python code for prime number": "Python code for prime number, and also the P8 and C, respectively.\nIn fact, the same rules apply to the P8 range, which is the same in both cases. In either case, we may make a higher number, with the P8 higher; in the case of the P8 lower one,",
210
+ "The best way to learn a new language is": "The best way to learn a new language is to read our language, and then get over to the basics. If you're ready to learn a new language, it's easy!\nWhat Do I Need to Know?\nYou are speaking to a native speaker every day, not on the same day, at the same time. That's right"
211
+ }
212
+ }
m22/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7b240badfbb7ac04b19cec038fa0728e85377114cd524c4d7cc3a8b4d17c6b4
3
+ size 256079264