B4K2xx commited on
Commit
30411c3
·
verified ·
1 Parent(s): 0a997cf

Add m20 weights and results

Browse files
m20/benchmark_results.json ADDED
@@ -0,0 +1,540 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "m20",
3
+ "checkpoint": "results/m20/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": 575.1581335575003,
13
+ "perplexity_stderr,none": 31.84994846752592,
14
+ "acc,none": 0.1847467494663303,
15
+ "acc_stderr,none": 0.005406880178651611
16
+ },
17
+ "hellaswag": {
18
+ "name": "hellaswag",
19
+ "alias": "hellaswag",
20
+ "sample_len": 10042,
21
+ "acc,none": 0.2711611232822147,
22
+ "acc_stderr,none": 0.004436505187567018,
23
+ "acc_norm,none": 0.2804222266480781,
24
+ "acc_norm_stderr,none": 0.004482874732237671
25
+ },
26
+ "piqa": {
27
+ "name": "piqa",
28
+ "alias": "piqa",
29
+ "sample_len": 1838,
30
+ "acc,none": 0.6050054406964092,
31
+ "acc_stderr,none": 0.011405665187969115,
32
+ "acc_norm,none": 0.5930359085963003,
33
+ "acc_norm_stderr,none": 0.011462093919190149
34
+ },
35
+ "sciq": {
36
+ "name": "sciq",
37
+ "alias": "sciq",
38
+ "sample_len": 1000,
39
+ "acc,none": 0.693,
40
+ "acc_stderr,none": 0.01459328489285274,
41
+ "acc_norm,none": 0.595,
42
+ "acc_norm_stderr,none": 0.01553113699045296
43
+ },
44
+ "arc_easy": {
45
+ "name": "arc_easy",
46
+ "alias": "arc_easy",
47
+ "sample_len": 2376,
48
+ "acc,none": 0.4524410774410774,
49
+ "acc_stderr,none": 0.010213265860171308,
50
+ "acc_norm,none": 0.41035353535353536,
51
+ "acc_norm_stderr,none": 0.01009353125576533
52
+ },
53
+ "winogrande": {
54
+ "name": "winogrande",
55
+ "alias": "winogrande",
56
+ "sample_len": 1267,
57
+ "acc,none": 0.5430149960536701,
58
+ "acc_stderr,none": 0.01400038676159808
59
+ },
60
+ "blimp_adjunct_island": {
61
+ "name": "blimp_adjunct_island",
62
+ "alias": "blimp_adjunct_island",
63
+ "sample_len": 1000,
64
+ "acc,none": 0.769,
65
+ "acc_stderr,none": 0.013334797216936478
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.921,
72
+ "acc_stderr,none": 0.00853415677333337
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.985,
79
+ "acc_stderr,none": 0.003845749574502972
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.741,
86
+ "acc_stderr,none": 0.013860415257527861
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.902,
93
+ "acc_stderr,none": 0.009406619184621247
94
+ },
95
+ "blimp_causative": {
96
+ "name": "blimp_causative",
97
+ "alias": "blimp_causative",
98
+ "sample_len": 1000,
99
+ "acc,none": 0.693,
100
+ "acc_stderr,none": 0.01459328489285274
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.365,
107
+ "acc_stderr,none": 0.015231776226264848
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.443,
114
+ "acc_stderr,none": 0.015716169953204184
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.763,
121
+ "acc_stderr,none": 0.013454070462577935
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.937,
135
+ "acc_stderr,none": 0.007687007876286461
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.903,
142
+ "acc_stderr,none": 0.009363689373248062
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.926,
149
+ "acc_stderr,none": 0.0082820645127041
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.917,
156
+ "acc_stderr,none": 0.008728527206074756
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.88,
163
+ "acc_stderr,none": 0.010281328012747462
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.899,
170
+ "acc_stderr,none": 0.009533618929341046
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.947,
177
+ "acc_stderr,none": 0.00708810561724649
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.85,
184
+ "acc_stderr,none": 0.011297239823409418
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.72,
191
+ "acc_stderr,none": 0.014205696104091548
192
+ },
193
+ "blimp_drop_argument": {
194
+ "name": "blimp_drop_argument",
195
+ "alias": "blimp_drop_argument",
196
+ "sample_len": 1000,
197
+ "acc,none": 0.749,
198
+ "acc_stderr,none": 0.013718133516888775
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.785,
205
+ "acc_stderr,none": 0.01299784381903183
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.843,
212
+ "acc_stderr,none": 0.011510146979230262
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.799,
219
+ "acc_stderr,none": 0.012679107214617274
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.957,
226
+ "acc_stderr,none": 0.006418114379799739
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.173,
233
+ "acc_stderr,none": 0.011967214137559927
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.868,
240
+ "acc_stderr,none": 0.010709373963527965
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.751,
247
+ "acc_stderr,none": 0.013681600278702275
248
+ },
249
+ "blimp_inchoative": {
250
+ "name": "blimp_inchoative",
251
+ "alias": "blimp_inchoative",
252
+ "sample_len": 1000,
253
+ "acc,none": 0.662,
254
+ "acc_stderr,none": 0.014965960710224385
255
+ },
256
+ "blimp_intransitive": {
257
+ "name": "blimp_intransitive",
258
+ "alias": "blimp_intransitive",
259
+ "sample_len": 1000,
260
+ "acc,none": 0.769,
261
+ "acc_stderr,none": 0.013334797216936478
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.955,
268
+ "acc_stderr,none": 0.006558812241406063
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.852,
275
+ "acc_stderr,none": 0.011234866364235145
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.886,
282
+ "acc_stderr,none": 0.010055103435823278
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.853,
289
+ "acc_stderr,none": 0.011203415395160376
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.53,
296
+ "acc_stderr,none": 0.015790799515836725
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.445,
303
+ "acc_stderr,none": 0.015723301886761007
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.619,
310
+ "acc_stderr,none": 0.015364734787007391
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.49,
317
+ "acc_stderr,none": 0.015816135752773183
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.643,
324
+ "acc_stderr,none": 0.01515852172148659
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.858,
331
+ "acc_stderr,none": 0.011043457699378305
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.611,
338
+ "acc_stderr,none": 0.015424555647308514
339
+ },
340
+ "blimp_passive_1": {
341
+ "name": "blimp_passive_1",
342
+ "alias": "blimp_passive_1",
343
+ "sample_len": 1000,
344
+ "acc,none": 0.889,
345
+ "acc_stderr,none": 0.009938701010583716
346
+ },
347
+ "blimp_passive_2": {
348
+ "name": "blimp_passive_2",
349
+ "alias": "blimp_passive_2",
350
+ "sample_len": 1000,
351
+ "acc,none": 0.875,
352
+ "acc_stderr,none": 0.010463483381956722
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.595,
359
+ "acc_stderr,none": 0.01553113699045296
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.929,
373
+ "acc_stderr,none": 0.008125578442487959
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.945,
380
+ "acc_stderr,none": 0.0072129762946392
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.723,
387
+ "acc_stderr,none": 0.014158794845306273
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.598,
394
+ "acc_stderr,none": 0.015512467135714959
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.38,
401
+ "acc_stderr,none": 0.015356947477797658
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.892,
408
+ "acc_stderr,none": 0.009820001651345708
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.997,
422
+ "acc_stderr,none": 0.0017303161543469432
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.555,
429
+ "acc_stderr,none": 0.015723301886761007
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.384,
436
+ "acc_stderr,none": 0.015387682761896958
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.924,
443
+ "acc_stderr,none": 0.008384169266796462
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.819,
450
+ "acc_stderr,none": 0.012181436179178027
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.641,
457
+ "acc_stderr,none": 0.01517726422479853
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.845,
464
+ "acc_stderr,none": 0.011450157470799522
465
+ },
466
+ "blimp_transitive": {
467
+ "name": "blimp_transitive",
468
+ "alias": "blimp_transitive",
469
+ "sample_len": 1000,
470
+ "acc,none": 0.804,
471
+ "acc_stderr,none": 0.012559527926707347
472
+ },
473
+ "blimp_wh_island": {
474
+ "name": "blimp_wh_island",
475
+ "alias": "blimp_wh_island",
476
+ "sample_len": 1000,
477
+ "acc,none": 0.67,
478
+ "acc_stderr,none": 0.014876872027456623
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.715,
485
+ "acc_stderr,none": 0.014282120955200562
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.921,
492
+ "acc_stderr,none": 0.00853415677333337
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.873,
499
+ "acc_stderr,none": 0.010534798620855644
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.963,
506
+ "acc_stderr,none": 0.0059721576223896395
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.955,
513
+ "acc_stderr,none": 0.006558812241406063
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.512,
520
+ "acc_stderr,none": 0.015814743314581686
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.283,
527
+ "acc_stderr,none": 0.01425181090648182
528
+ },
529
+ "blimp": {
530
+ "alias": "blimp",
531
+ "name": "blimp",
532
+ "sample_len": 67000,
533
+ "acc,none": 0.7641641791044775,
534
+ "acc_stderr,none": 0.0014648618904583065,
535
+ "sample_count": {
536
+ "acc,none": 67000
537
+ }
538
+ }
539
+ }
540
+ }
m20/config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "m20",
3
+ "run_name": "m20-mla-rope-latent256-dense-adamw",
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": "mla",
16
+ "gqa_num_kv_heads": 2,
17
+ "mla_latent_dim": 256,
18
+ "pos_type": "rope",
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": 1337,
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
+ }
m20/eval_results.json ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "m20",
3
+ "checkpoint": "/home/akshat/codemindv2/results/m20/final.pt",
4
+ "trained_step": 15258,
5
+ "total_params": 61920256,
6
+ "active_params": 61920256,
7
+ "flops_per_forward": 123840512.0,
8
+ "val_loss": 3.4625886261463164,
9
+ "val_perplexity": 31.899445466560927,
10
+ "bits_per_token": 4.99545943957982,
11
+ "ppl_tokens_evaluated": 655360,
12
+ "prefill_tok_per_sec": 158358.59954810305,
13
+ "inference_tok_per_sec": 178.26855692268356,
14
+ "inference_tok_per_sec_kvcache": 191.98823619444323,
15
+ "kvcache_speedup": 1.076960735581149,
16
+ "gen_tokens": 128,
17
+ "peak_infer_vram_gb": 0.385,
18
+ "peak_train_vram_gb": 4.062,
19
+ "closeness_answer_ppl": 23.819,
20
+ "closeness_hit_rate": 0.45,
21
+ "closeness_token_f1": 0.014,
22
+ "closeness_n": 20,
23
+ "closeness_items": [
24
+ {
25
+ "prompt": "The capital of France is",
26
+ "answer": " Paris",
27
+ "generated": " the city of Paris.\n",
28
+ "answer_nll": 2.9506,
29
+ "answer_ppl": 19.117,
30
+ "hit": true,
31
+ "token_f1": 0.0
32
+ },
33
+ {
34
+ "prompt": "The capital of Japan is",
35
+ "answer": " Tokyo",
36
+ "generated": " the capital of the Republic of",
37
+ "answer_nll": 2.6959,
38
+ "answer_ppl": 14.818,
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": 2.8111,
47
+ "answer_ppl": 16.628,
48
+ "hit": false,
49
+ "token_f1": 0.0
50
+ },
51
+ {
52
+ "prompt": "Water is made of hydrogen and",
53
+ "answer": " oxygen",
54
+ "generated": " oxygen.\nThe hydrogen is",
55
+ "answer_nll": 0.7412,
56
+ "answer_ppl": 2.099,
57
+ "hit": true,
58
+ "token_f1": 0.0
59
+ },
60
+ {
61
+ "prompt": "The opposite of hot is",
62
+ "answer": " cold",
63
+ "generated": " the case for the two-",
64
+ "answer_nll": 5.2195,
65
+ "answer_ppl": 184.842,
66
+ "hit": false,
67
+ "token_f1": 0.0
68
+ },
69
+ {
70
+ "prompt": "The opposite of up is",
71
+ "answer": " down",
72
+ "generated": " the case of the two-",
73
+ "answer_nll": 7.8645,
74
+ "answer_ppl": 2603.199,
75
+ "hit": false,
76
+ "token_f1": 0.0
77
+ },
78
+ {
79
+ "prompt": "Two plus two equals",
80
+ "answer": " four",
81
+ "generated": " one.\nThe first two",
82
+ "answer_nll": 3.4823,
83
+ "answer_ppl": 32.534,
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.5493,
92
+ "answer_ppl": 34.79,
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.\nThe first president",
100
+ "answer_nll": 0.3028,
101
+ "answer_ppl": 1.354,
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.3426,
110
+ "answer_ppl": 1.409,
111
+ "hit": true,
112
+ "token_f1": 0.0
113
+ },
114
+ {
115
+ "prompt": "A triangle has three",
116
+ "answer": " sides",
117
+ "generated": " sides.\n- The triangle",
118
+ "answer_nll": 1.2375,
119
+ "answer_ppl": 3.447,
120
+ "hit": true,
121
+ "token_f1": 0.0
122
+ },
123
+ {
124
+ "prompt": "The chemical symbol for gold is",
125
+ "answer": " Au",
126
+ "generated": " gold.\nThe gold symbol",
127
+ "answer_nll": 9.1208,
128
+ "answer_ppl": 9143.113,
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": 4.3002,
137
+ "answer_ppl": 73.711,
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": 0.1833,
146
+ "answer_ppl": 1.201,
147
+ "hit": true,
148
+ "token_f1": 0.0
149
+ },
150
+ {
151
+ "prompt": "The Earth revolves around the",
152
+ "answer": " Sun",
153
+ "generated": " sun.\nThe Earth is",
154
+ "answer_nll": 0.9723,
155
+ "answer_ppl": 2.644,
156
+ "hit": true,
157
+ "token_f1": 0.0
158
+ },
159
+ {
160
+ "prompt": "The plural of mouse is",
161
+ "answer": " mice",
162
+ "generated": " the plural of mouse.\n",
163
+ "answer_nll": 7.0402,
164
+ "answer_ppl": 1141.636,
165
+ "hit": false,
166
+ "token_f1": 0.0
167
+ },
168
+ {
169
+ "prompt": "Photosynthesis occurs in the",
170
+ "answer": " leaves",
171
+ "generated": " form of photosynthesis.\n",
172
+ "answer_nll": 3.8898,
173
+ "answer_ppl": 48.901,
174
+ "hit": false,
175
+ "token_f1": 0.0
176
+ },
177
+ {
178
+ "prompt": "The human heart pumps",
179
+ "answer": " blood",
180
+ "generated": " blood and oxygen to the brain",
181
+ "answer_nll": 1.0091,
182
+ "answer_ppl": 2.743,
183
+ "hit": true,
184
+ "token_f1": 0.286
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": 2.9926,
191
+ "answer_ppl": 19.938,
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": " elephant.\nThe elephant is",
199
+ "answer_nll": 2.7043,
200
+ "answer_ppl": 14.944,
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 to look at your own body and how you feel.\n- Try to get out of bed every night.\n- Do you feel sleepy all the time?\n- Do you feel like you are having a bad day, no matter what the time?\n- Do you feel as though you can",
207
+ "Scientists have discovered that": "Scientists have discovered that, in a recent study of about 30,000 participants from the Australian and European Union, the amount of sunlight in the atmosphere of the planet is about one-third of the sun\u2019s mass.\n\u201cWe are seeing significant changes in the intensity of light, but the effects this type",
208
+ "In the history of mathematics,": "In the history of mathematics, the term \"physics\" was used to describe the operations of many different fields.\nThe science of physics has become more and more popular in recent years, with the number of scientific journals running, and the number of experimental physicists working on them. There are numerous other fields, and their roles vary",
209
+ "Python code for prime number": "Python code for prime number of integers.\n- The integer values of the integers for integers are integers and integers.\n- When the integers are integers, the integers are integers and integers.\n- The integer values of the integers are integers and integers.\n- The integers are integers\n- The integers are integers\n-",
210
+ "The best way to learn a new language is": "The best way to learn a new language is to learn about its features and benefits. It is especially important to study the language in a way that is appropriate for your child's age.\nIt is very important to keep in mind that the best way to learn a language is to learn some basic elements. For example, learn to write a sentence"
211
+ }
212
+ }
m20/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65703dd98a9c3afa608d2a7d09a83363b4f8cb7bf29c1a2c56826aee4b61d468
3
+ size 247691704