YagiASAFAS commited on
Commit
6457906
·
verified ·
1 Parent(s): f423488

Training in progress, epoch 1

Browse files
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:26d45660e79eae18a1d64284246777a584fb8ee16373738db06275d022d6d302
3
  size 438007864
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52c4ecf0fb267ca0111315770c7aee028de79177925924f83b69d961b456b023
3
  size 438007864
run-0/checkpoint-10000/config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3",
16
+ "4": "LABEL_4",
17
+ "5": "LABEL_5",
18
+ "6": "LABEL_6",
19
+ "7": "LABEL_7",
20
+ "8": "LABEL_8",
21
+ "9": "LABEL_9",
22
+ "10": "LABEL_10",
23
+ "11": "LABEL_11",
24
+ "12": "LABEL_12",
25
+ "13": "LABEL_13",
26
+ "14": "LABEL_14",
27
+ "15": "LABEL_15",
28
+ "16": "LABEL_16",
29
+ "17": "LABEL_17"
30
+ },
31
+ "initializer_range": 0.02,
32
+ "intermediate_size": 3072,
33
+ "label2id": {
34
+ "LABEL_0": 0,
35
+ "LABEL_1": 1,
36
+ "LABEL_10": 10,
37
+ "LABEL_11": 11,
38
+ "LABEL_12": 12,
39
+ "LABEL_13": 13,
40
+ "LABEL_14": 14,
41
+ "LABEL_15": 15,
42
+ "LABEL_16": 16,
43
+ "LABEL_17": 17,
44
+ "LABEL_2": 2,
45
+ "LABEL_3": 3,
46
+ "LABEL_4": 4,
47
+ "LABEL_5": 5,
48
+ "LABEL_6": 6,
49
+ "LABEL_7": 7,
50
+ "LABEL_8": 8,
51
+ "LABEL_9": 9
52
+ },
53
+ "layer_norm_eps": 1e-12,
54
+ "max_position_embeddings": 512,
55
+ "model_type": "bert",
56
+ "num_attention_heads": 12,
57
+ "num_hidden_layers": 12,
58
+ "pad_token_id": 0,
59
+ "position_embedding_type": "absolute",
60
+ "torch_dtype": "float32",
61
+ "transformers_version": "4.55.4",
62
+ "type_vocab_size": 2,
63
+ "use_cache": true,
64
+ "vocab_size": 30522
65
+ }
run-0/checkpoint-10000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:151d50ff031b36f785718e9e9f316f3d7a5e87e2f11599c57ed44c7c4d9a8527
3
+ size 438007864
run-0/checkpoint-10000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a66ad6b533a76a6bfdeec0d870a00afebaa861b50fbcc74ebb68ef6df365f29a
3
+ size 876140043
run-0/checkpoint-10000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14c39e64e59e8286453bf0b17c32f65ff9b3c161db8386dcfa997c86bd201138
3
+ size 14645
run-0/checkpoint-10000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb74b75bedf8782dea7efd25c13f85c7a8962f335393852b7cabaca23984beb0
3
+ size 1383
run-0/checkpoint-10000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a862f187509064124ca19d151ecafb5a63f35c3823a6203b8fc6a3956261cf5d
3
+ size 1465
run-0/checkpoint-10000/trainer_state.json ADDED
@@ -0,0 +1,860 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 10000,
3
+ "best_metric": 0.8275137305036369,
4
+ "best_model_checkpoint": "./results/run-0/checkpoint-10000",
5
+ "epoch": 5.0,
6
+ "eval_steps": 500,
7
+ "global_step": 10000,
8
+ "is_hyper_param_search": true,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.05,
14
+ "grad_norm": 2.039576768875122,
15
+ "learning_rate": 1.447537100215313e-06,
16
+ "loss": 0.0,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 0.1,
21
+ "grad_norm": 4.460484504699707,
22
+ "learning_rate": 2.9096957873014878e-06,
23
+ "loss": 0.0821,
24
+ "step": 200
25
+ },
26
+ {
27
+ "epoch": 0.15,
28
+ "grad_norm": 26.84711456298828,
29
+ "learning_rate": 4.371854474387663e-06,
30
+ "loss": 0.0519,
31
+ "step": 300
32
+ },
33
+ {
34
+ "epoch": 0.2,
35
+ "grad_norm": 0.6646966934204102,
36
+ "learning_rate": 5.834013161473838e-06,
37
+ "loss": 0.0,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 0.25,
42
+ "grad_norm": 1.8385223150253296,
43
+ "learning_rate": 7.296171848560012e-06,
44
+ "loss": 0.0,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.3,
49
+ "grad_norm": 0.9234638810157776,
50
+ "learning_rate": 8.758330535646188e-06,
51
+ "loss": 0.0,
52
+ "step": 600
53
+ },
54
+ {
55
+ "epoch": 0.35,
56
+ "grad_norm": 38.24382400512695,
57
+ "learning_rate": 9.9933989317167e-06,
58
+ "loss": 0.0,
59
+ "step": 700
60
+ },
61
+ {
62
+ "epoch": 0.4,
63
+ "grad_norm": 42.83884811401367,
64
+ "learning_rate": 9.94162234536513e-06,
65
+ "loss": 0.0,
66
+ "step": 800
67
+ },
68
+ {
69
+ "epoch": 0.45,
70
+ "grad_norm": 0.8020399808883667,
71
+ "learning_rate": 9.889845759013559e-06,
72
+ "loss": 0.0,
73
+ "step": 900
74
+ },
75
+ {
76
+ "epoch": 0.5,
77
+ "grad_norm": 0.86091148853302,
78
+ "learning_rate": 9.838069172661988e-06,
79
+ "loss": 0.0,
80
+ "step": 1000
81
+ },
82
+ {
83
+ "epoch": 0.55,
84
+ "grad_norm": 0.16439934074878693,
85
+ "learning_rate": 9.786292586310415e-06,
86
+ "loss": 0.0,
87
+ "step": 1100
88
+ },
89
+ {
90
+ "epoch": 0.6,
91
+ "grad_norm": 2.752115488052368,
92
+ "learning_rate": 9.734515999958846e-06,
93
+ "loss": 0.0,
94
+ "step": 1200
95
+ },
96
+ {
97
+ "epoch": 0.65,
98
+ "grad_norm": 1.5777802467346191,
99
+ "learning_rate": 9.682739413607275e-06,
100
+ "loss": 0.0,
101
+ "step": 1300
102
+ },
103
+ {
104
+ "epoch": 0.7,
105
+ "grad_norm": 5.708035945892334,
106
+ "learning_rate": 9.630962827255703e-06,
107
+ "loss": 0.0598,
108
+ "step": 1400
109
+ },
110
+ {
111
+ "epoch": 0.75,
112
+ "grad_norm": 0.3160113990306854,
113
+ "learning_rate": 9.579186240904132e-06,
114
+ "loss": 0.0441,
115
+ "step": 1500
116
+ },
117
+ {
118
+ "epoch": 0.8,
119
+ "grad_norm": 1.2273201942443848,
120
+ "learning_rate": 9.527409654552563e-06,
121
+ "loss": 0.0,
122
+ "step": 1600
123
+ },
124
+ {
125
+ "epoch": 0.85,
126
+ "grad_norm": 29.215551376342773,
127
+ "learning_rate": 9.47563306820099e-06,
128
+ "loss": 0.0,
129
+ "step": 1700
130
+ },
131
+ {
132
+ "epoch": 0.9,
133
+ "grad_norm": 0.18806450068950653,
134
+ "learning_rate": 9.42385648184942e-06,
135
+ "loss": 0.0,
136
+ "step": 1800
137
+ },
138
+ {
139
+ "epoch": 0.95,
140
+ "grad_norm": 0.41723617911338806,
141
+ "learning_rate": 9.372079895497849e-06,
142
+ "loss": 0.0183,
143
+ "step": 1900
144
+ },
145
+ {
146
+ "epoch": 1.0,
147
+ "grad_norm": 1.9550623893737793,
148
+ "learning_rate": 9.32030330914628e-06,
149
+ "loss": 0.0,
150
+ "step": 2000
151
+ },
152
+ {
153
+ "epoch": 1.0,
154
+ "eval_economic_inequality_accuracy": 0.8047858942065491,
155
+ "eval_economic_inequality_f1": 0.7536914695213144,
156
+ "eval_economic_policy_benefits_accuracy": 0.8282828282828283,
157
+ "eval_economic_policy_benefits_f1": 0.7567090529234769,
158
+ "eval_ethnic_boundaries_accuracy": 0.9526143790849673,
159
+ "eval_ethnic_boundaries_f1": 0.9344984933861545,
160
+ "eval_language_policy_accuracy": 0.7058823529411765,
161
+ "eval_language_policy_f1": 0.6071428571428571,
162
+ "eval_loss": NaN,
163
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
164
+ "eval_mother_tongue_education_f1": 0.837037037037037,
165
+ "eval_overall_accuracy": 0.8392311075242039,
166
+ "eval_overall_f1": 0.782435168791527,
167
+ "eval_religion_ethnic_identity_accuracy": 0.8549323017408124,
168
+ "eval_religion_ethnic_identity_f1": 0.8055321027383222,
169
+ "eval_runtime": 4.4874,
170
+ "eval_samples_per_second": 891.378,
171
+ "eval_steps_per_second": 55.711,
172
+ "step": 2000
173
+ },
174
+ {
175
+ "epoch": 1.05,
176
+ "grad_norm": 2.762986183166504,
177
+ "learning_rate": 9.268526722794707e-06,
178
+ "loss": 0.0,
179
+ "step": 2100
180
+ },
181
+ {
182
+ "epoch": 1.1,
183
+ "grad_norm": 0.5487369298934937,
184
+ "learning_rate": 9.216750136443136e-06,
185
+ "loss": 0.0,
186
+ "step": 2200
187
+ },
188
+ {
189
+ "epoch": 1.15,
190
+ "grad_norm": 4.76751184463501,
191
+ "learning_rate": 9.164973550091565e-06,
192
+ "loss": 0.0102,
193
+ "step": 2300
194
+ },
195
+ {
196
+ "epoch": 1.2,
197
+ "grad_norm": 2.681756019592285,
198
+ "learning_rate": 9.113196963739995e-06,
199
+ "loss": 0.0,
200
+ "step": 2400
201
+ },
202
+ {
203
+ "epoch": 1.25,
204
+ "grad_norm": 0.24124185740947723,
205
+ "learning_rate": 9.061420377388424e-06,
206
+ "loss": 0.0,
207
+ "step": 2500
208
+ },
209
+ {
210
+ "epoch": 1.3,
211
+ "grad_norm": 1.943154215812683,
212
+ "learning_rate": 9.009643791036853e-06,
213
+ "loss": 0.0,
214
+ "step": 2600
215
+ },
216
+ {
217
+ "epoch": 1.35,
218
+ "grad_norm": 0.28053101897239685,
219
+ "learning_rate": 8.957867204685282e-06,
220
+ "loss": 0.0347,
221
+ "step": 2700
222
+ },
223
+ {
224
+ "epoch": 1.4,
225
+ "grad_norm": 1.7952568531036377,
226
+ "learning_rate": 8.906090618333711e-06,
227
+ "loss": 0.4436,
228
+ "step": 2800
229
+ },
230
+ {
231
+ "epoch": 1.45,
232
+ "grad_norm": 3.859933614730835,
233
+ "learning_rate": 8.85431403198214e-06,
234
+ "loss": 0.0,
235
+ "step": 2900
236
+ },
237
+ {
238
+ "epoch": 1.5,
239
+ "grad_norm": 1.094464898109436,
240
+ "learning_rate": 8.80253744563057e-06,
241
+ "loss": 0.0,
242
+ "step": 3000
243
+ },
244
+ {
245
+ "epoch": 1.55,
246
+ "grad_norm": 0.5670285820960999,
247
+ "learning_rate": 8.750760859278999e-06,
248
+ "loss": 0.0,
249
+ "step": 3100
250
+ },
251
+ {
252
+ "epoch": 1.6,
253
+ "grad_norm": 2.8412418365478516,
254
+ "learning_rate": 8.698984272927428e-06,
255
+ "loss": 0.0,
256
+ "step": 3200
257
+ },
258
+ {
259
+ "epoch": 1.65,
260
+ "grad_norm": 65.82475280761719,
261
+ "learning_rate": 8.647207686575855e-06,
262
+ "loss": 0.0,
263
+ "step": 3300
264
+ },
265
+ {
266
+ "epoch": 1.7,
267
+ "grad_norm": 8.976615905761719,
268
+ "learning_rate": 8.595431100224286e-06,
269
+ "loss": 0.0048,
270
+ "step": 3400
271
+ },
272
+ {
273
+ "epoch": 1.75,
274
+ "grad_norm": 126.71548461914062,
275
+ "learning_rate": 8.543654513872716e-06,
276
+ "loss": 0.0,
277
+ "step": 3500
278
+ },
279
+ {
280
+ "epoch": 1.8,
281
+ "grad_norm": 0.595015823841095,
282
+ "learning_rate": 8.491877927521143e-06,
283
+ "loss": 0.0033,
284
+ "step": 3600
285
+ },
286
+ {
287
+ "epoch": 1.85,
288
+ "grad_norm": 3.98215651512146,
289
+ "learning_rate": 8.440101341169572e-06,
290
+ "loss": 0.1482,
291
+ "step": 3700
292
+ },
293
+ {
294
+ "epoch": 1.9,
295
+ "grad_norm": 3.0417540073394775,
296
+ "learning_rate": 8.388324754818003e-06,
297
+ "loss": 0.0031,
298
+ "step": 3800
299
+ },
300
+ {
301
+ "epoch": 1.95,
302
+ "grad_norm": 27.85649299621582,
303
+ "learning_rate": 8.33654816846643e-06,
304
+ "loss": 0.0589,
305
+ "step": 3900
306
+ },
307
+ {
308
+ "epoch": 2.0,
309
+ "grad_norm": 2.3094232082366943,
310
+ "learning_rate": 8.28477158211486e-06,
311
+ "loss": 0.0,
312
+ "step": 4000
313
+ },
314
+ {
315
+ "epoch": 2.0,
316
+ "eval_economic_inequality_accuracy": 0.8236775818639799,
317
+ "eval_economic_inequality_f1": 0.8161521311449569,
318
+ "eval_economic_policy_benefits_accuracy": 0.835016835016835,
319
+ "eval_economic_policy_benefits_f1": 0.81117677081242,
320
+ "eval_ethnic_boundaries_accuracy": 0.9477124183006536,
321
+ "eval_ethnic_boundaries_f1": 0.9421262334088456,
322
+ "eval_language_policy_accuracy": 0.7058823529411765,
323
+ "eval_language_policy_f1": 0.6071428571428571,
324
+ "eval_loss": NaN,
325
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
326
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
327
+ "eval_overall_accuracy": 0.8277126438976526,
328
+ "eval_overall_f1": 0.7960173130694123,
329
+ "eval_religion_ethnic_identity_accuracy": 0.8762088974854932,
330
+ "eval_religion_ethnic_identity_f1": 0.8587651451666537,
331
+ "eval_runtime": 4.4654,
332
+ "eval_samples_per_second": 895.775,
333
+ "eval_steps_per_second": 55.986,
334
+ "step": 4000
335
+ },
336
+ {
337
+ "epoch": 2.05,
338
+ "grad_norm": 8.547677040100098,
339
+ "learning_rate": 8.232994995763289e-06,
340
+ "loss": 0.0,
341
+ "step": 4100
342
+ },
343
+ {
344
+ "epoch": 2.1,
345
+ "grad_norm": 117.06266784667969,
346
+ "learning_rate": 8.181218409411718e-06,
347
+ "loss": 0.0,
348
+ "step": 4200
349
+ },
350
+ {
351
+ "epoch": 2.15,
352
+ "grad_norm": 0.19274334609508514,
353
+ "learning_rate": 8.129441823060147e-06,
354
+ "loss": 0.0,
355
+ "step": 4300
356
+ },
357
+ {
358
+ "epoch": 2.2,
359
+ "grad_norm": 1.0360257625579834,
360
+ "learning_rate": 8.077665236708576e-06,
361
+ "loss": 0.0,
362
+ "step": 4400
363
+ },
364
+ {
365
+ "epoch": 2.25,
366
+ "grad_norm": 0.08476172387599945,
367
+ "learning_rate": 8.025888650357006e-06,
368
+ "loss": 0.0,
369
+ "step": 4500
370
+ },
371
+ {
372
+ "epoch": 2.3,
373
+ "grad_norm": 0.08015169203281403,
374
+ "learning_rate": 7.974112064005435e-06,
375
+ "loss": 0.0,
376
+ "step": 4600
377
+ },
378
+ {
379
+ "epoch": 2.35,
380
+ "grad_norm": 0.22844159603118896,
381
+ "learning_rate": 7.922335477653864e-06,
382
+ "loss": 0.0006,
383
+ "step": 4700
384
+ },
385
+ {
386
+ "epoch": 2.4,
387
+ "grad_norm": 0.08015509694814682,
388
+ "learning_rate": 7.870558891302293e-06,
389
+ "loss": 0.0006,
390
+ "step": 4800
391
+ },
392
+ {
393
+ "epoch": 2.45,
394
+ "grad_norm": 5.426948547363281,
395
+ "learning_rate": 7.818782304950722e-06,
396
+ "loss": 0.0,
397
+ "step": 4900
398
+ },
399
+ {
400
+ "epoch": 2.5,
401
+ "grad_norm": 0.022730231285095215,
402
+ "learning_rate": 7.767005718599151e-06,
403
+ "loss": 0.0,
404
+ "step": 5000
405
+ },
406
+ {
407
+ "epoch": 2.55,
408
+ "grad_norm": 5.662137508392334,
409
+ "learning_rate": 7.71522913224758e-06,
410
+ "loss": 0.0,
411
+ "step": 5100
412
+ },
413
+ {
414
+ "epoch": 2.6,
415
+ "grad_norm": 0.05326724052429199,
416
+ "learning_rate": 7.66345254589601e-06,
417
+ "loss": 0.0,
418
+ "step": 5200
419
+ },
420
+ {
421
+ "epoch": 2.65,
422
+ "grad_norm": 2.7219042778015137,
423
+ "learning_rate": 7.611675959544438e-06,
424
+ "loss": 0.0003,
425
+ "step": 5300
426
+ },
427
+ {
428
+ "epoch": 2.7,
429
+ "grad_norm": 4.045983791351318,
430
+ "learning_rate": 7.559899373192868e-06,
431
+ "loss": 0.0,
432
+ "step": 5400
433
+ },
434
+ {
435
+ "epoch": 2.75,
436
+ "grad_norm": 0.17610976099967957,
437
+ "learning_rate": 7.5081227868412964e-06,
438
+ "loss": 0.0,
439
+ "step": 5500
440
+ },
441
+ {
442
+ "epoch": 2.8,
443
+ "grad_norm": 0.03921931982040405,
444
+ "learning_rate": 7.456346200489726e-06,
445
+ "loss": 0.0,
446
+ "step": 5600
447
+ },
448
+ {
449
+ "epoch": 2.85,
450
+ "grad_norm": 3.3493025302886963,
451
+ "learning_rate": 7.404569614138155e-06,
452
+ "loss": 0.0,
453
+ "step": 5700
454
+ },
455
+ {
456
+ "epoch": 2.9,
457
+ "grad_norm": 0.17839455604553223,
458
+ "learning_rate": 7.352793027786584e-06,
459
+ "loss": 0.0,
460
+ "step": 5800
461
+ },
462
+ {
463
+ "epoch": 2.95,
464
+ "grad_norm": 0.01533100288361311,
465
+ "learning_rate": 7.301016441435013e-06,
466
+ "loss": 0.0,
467
+ "step": 5900
468
+ },
469
+ {
470
+ "epoch": 3.0,
471
+ "grad_norm": 0.6168579459190369,
472
+ "learning_rate": 7.249239855083442e-06,
473
+ "loss": 0.0,
474
+ "step": 6000
475
+ },
476
+ {
477
+ "epoch": 3.0,
478
+ "eval_economic_inequality_accuracy": 0.818639798488665,
479
+ "eval_economic_inequality_f1": 0.8128041981528128,
480
+ "eval_economic_policy_benefits_accuracy": 0.8686868686868687,
481
+ "eval_economic_policy_benefits_f1": 0.849679269639744,
482
+ "eval_ethnic_boundaries_accuracy": 0.9379084967320261,
483
+ "eval_ethnic_boundaries_f1": 0.9377319200237987,
484
+ "eval_language_policy_accuracy": 0.6470588235294118,
485
+ "eval_language_policy_f1": 0.5673202614379085,
486
+ "eval_loss": NaN,
487
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
488
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
489
+ "eval_overall_accuracy": 0.8181454231515234,
490
+ "eval_overall_f1": 0.794560338168735,
491
+ "eval_religion_ethnic_identity_accuracy": 0.8588007736943907,
492
+ "eval_religion_ethnic_identity_f1": 0.8590856390174055,
493
+ "eval_runtime": 4.4722,
494
+ "eval_samples_per_second": 894.406,
495
+ "eval_steps_per_second": 55.9,
496
+ "step": 6000
497
+ },
498
+ {
499
+ "epoch": 3.05,
500
+ "grad_norm": 0.06797665357589722,
501
+ "learning_rate": 7.197463268731871e-06,
502
+ "loss": 0.0,
503
+ "step": 6100
504
+ },
505
+ {
506
+ "epoch": 3.1,
507
+ "grad_norm": 0.05142182484269142,
508
+ "learning_rate": 7.145686682380301e-06,
509
+ "loss": 0.0002,
510
+ "step": 6200
511
+ },
512
+ {
513
+ "epoch": 3.15,
514
+ "grad_norm": 0.04868694022297859,
515
+ "learning_rate": 7.09391009602873e-06,
516
+ "loss": 0.0,
517
+ "step": 6300
518
+ },
519
+ {
520
+ "epoch": 3.2,
521
+ "grad_norm": 0.026197949424386024,
522
+ "learning_rate": 7.042133509677158e-06,
523
+ "loss": 0.0005,
524
+ "step": 6400
525
+ },
526
+ {
527
+ "epoch": 3.25,
528
+ "grad_norm": 12.894441604614258,
529
+ "learning_rate": 6.990356923325587e-06,
530
+ "loss": 0.0,
531
+ "step": 6500
532
+ },
533
+ {
534
+ "epoch": 3.3,
535
+ "grad_norm": 4.21354341506958,
536
+ "learning_rate": 6.938580336974017e-06,
537
+ "loss": 0.0,
538
+ "step": 6600
539
+ },
540
+ {
541
+ "epoch": 3.35,
542
+ "grad_norm": 0.0193119365721941,
543
+ "learning_rate": 6.886803750622446e-06,
544
+ "loss": 0.0,
545
+ "step": 6700
546
+ },
547
+ {
548
+ "epoch": 3.4,
549
+ "grad_norm": 0.07511546462774277,
550
+ "learning_rate": 6.835027164270875e-06,
551
+ "loss": 0.0,
552
+ "step": 6800
553
+ },
554
+ {
555
+ "epoch": 3.45,
556
+ "grad_norm": 0.9280033111572266,
557
+ "learning_rate": 6.783250577919304e-06,
558
+ "loss": 0.0003,
559
+ "step": 6900
560
+ },
561
+ {
562
+ "epoch": 3.5,
563
+ "grad_norm": 108.68927001953125,
564
+ "learning_rate": 6.731473991567734e-06,
565
+ "loss": 0.0,
566
+ "step": 7000
567
+ },
568
+ {
569
+ "epoch": 3.55,
570
+ "grad_norm": 4.475680828094482,
571
+ "learning_rate": 6.679697405216162e-06,
572
+ "loss": 0.0,
573
+ "step": 7100
574
+ },
575
+ {
576
+ "epoch": 3.6,
577
+ "grad_norm": 0.040891651064157486,
578
+ "learning_rate": 6.6279208188645915e-06,
579
+ "loss": 0.0,
580
+ "step": 7200
581
+ },
582
+ {
583
+ "epoch": 3.65,
584
+ "grad_norm": 0.032305650413036346,
585
+ "learning_rate": 6.576144232513021e-06,
586
+ "loss": 0.0,
587
+ "step": 7300
588
+ },
589
+ {
590
+ "epoch": 3.7,
591
+ "grad_norm": 0.027986843138933182,
592
+ "learning_rate": 6.524367646161449e-06,
593
+ "loss": 0.0002,
594
+ "step": 7400
595
+ },
596
+ {
597
+ "epoch": 3.75,
598
+ "grad_norm": 0.023173924535512924,
599
+ "learning_rate": 6.472591059809879e-06,
600
+ "loss": 0.0,
601
+ "step": 7500
602
+ },
603
+ {
604
+ "epoch": 3.8,
605
+ "grad_norm": 0.10551729053258896,
606
+ "learning_rate": 6.420814473458308e-06,
607
+ "loss": 0.0,
608
+ "step": 7600
609
+ },
610
+ {
611
+ "epoch": 3.85,
612
+ "grad_norm": 4.631871223449707,
613
+ "learning_rate": 6.3690378871067366e-06,
614
+ "loss": 0.0,
615
+ "step": 7700
616
+ },
617
+ {
618
+ "epoch": 3.9,
619
+ "grad_norm": 0.09392645955085754,
620
+ "learning_rate": 6.317261300755166e-06,
621
+ "loss": 0.0,
622
+ "step": 7800
623
+ },
624
+ {
625
+ "epoch": 3.95,
626
+ "grad_norm": 0.32573947310447693,
627
+ "learning_rate": 6.265484714403596e-06,
628
+ "loss": 0.0,
629
+ "step": 7900
630
+ },
631
+ {
632
+ "epoch": 4.0,
633
+ "grad_norm": 0.012679875828325748,
634
+ "learning_rate": 6.213708128052024e-06,
635
+ "loss": 0.0,
636
+ "step": 8000
637
+ },
638
+ {
639
+ "epoch": 4.0,
640
+ "eval_economic_inequality_accuracy": 0.809823677581864,
641
+ "eval_economic_inequality_f1": 0.8053367188692558,
642
+ "eval_economic_policy_benefits_accuracy": 0.8451178451178452,
643
+ "eval_economic_policy_benefits_f1": 0.8262377279238239,
644
+ "eval_ethnic_boundaries_accuracy": 0.9379084967320261,
645
+ "eval_ethnic_boundaries_f1": 0.9368969810146281,
646
+ "eval_language_policy_accuracy": 0.7058823529411765,
647
+ "eval_language_policy_f1": 0.6071428571428571,
648
+ "eval_loss": NaN,
649
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
650
+ "eval_mother_tongue_education_f1": 0.837037037037037,
651
+ "eval_overall_accuracy": 0.8436493204617509,
652
+ "eval_overall_f1": 0.8135939571355589,
653
+ "eval_religion_ethnic_identity_accuracy": 0.874274661508704,
654
+ "eval_religion_ethnic_identity_f1": 0.8689124208257524,
655
+ "eval_runtime": 4.4538,
656
+ "eval_samples_per_second": 898.1,
657
+ "eval_steps_per_second": 56.131,
658
+ "step": 8000
659
+ },
660
+ {
661
+ "epoch": 4.05,
662
+ "grad_norm": 0.08179322630167007,
663
+ "learning_rate": 6.161931541700453e-06,
664
+ "loss": 0.0,
665
+ "step": 8100
666
+ },
667
+ {
668
+ "epoch": 4.1,
669
+ "grad_norm": 1.8278212547302246,
670
+ "learning_rate": 6.1101549553488824e-06,
671
+ "loss": 0.0,
672
+ "step": 8200
673
+ },
674
+ {
675
+ "epoch": 4.15,
676
+ "grad_norm": 0.9412699341773987,
677
+ "learning_rate": 6.058378368997311e-06,
678
+ "loss": 0.0,
679
+ "step": 8300
680
+ },
681
+ {
682
+ "epoch": 4.2,
683
+ "grad_norm": 0.3116857707500458,
684
+ "learning_rate": 6.006601782645741e-06,
685
+ "loss": 0.0,
686
+ "step": 8400
687
+ },
688
+ {
689
+ "epoch": 4.25,
690
+ "grad_norm": 0.04464244097471237,
691
+ "learning_rate": 5.95482519629417e-06,
692
+ "loss": 0.0,
693
+ "step": 8500
694
+ },
695
+ {
696
+ "epoch": 4.3,
697
+ "grad_norm": 0.0,
698
+ "learning_rate": 5.903048609942598e-06,
699
+ "loss": 0.0,
700
+ "step": 8600
701
+ },
702
+ {
703
+ "epoch": 4.35,
704
+ "grad_norm": 0.011906923726201057,
705
+ "learning_rate": 5.8512720235910274e-06,
706
+ "loss": 0.0,
707
+ "step": 8700
708
+ },
709
+ {
710
+ "epoch": 4.4,
711
+ "grad_norm": 0.022279787808656693,
712
+ "learning_rate": 5.7994954372394575e-06,
713
+ "loss": 0.0002,
714
+ "step": 8800
715
+ },
716
+ {
717
+ "epoch": 4.45,
718
+ "grad_norm": 0.07576213777065277,
719
+ "learning_rate": 5.747718850887886e-06,
720
+ "loss": 0.0001,
721
+ "step": 8900
722
+ },
723
+ {
724
+ "epoch": 4.5,
725
+ "grad_norm": 0.02016213722527027,
726
+ "learning_rate": 5.695942264536315e-06,
727
+ "loss": 0.0,
728
+ "step": 9000
729
+ },
730
+ {
731
+ "epoch": 4.55,
732
+ "grad_norm": 0.07930700480937958,
733
+ "learning_rate": 5.644165678184744e-06,
734
+ "loss": 0.0,
735
+ "step": 9100
736
+ },
737
+ {
738
+ "epoch": 4.6,
739
+ "grad_norm": 0.016734231263399124,
740
+ "learning_rate": 5.592389091833174e-06,
741
+ "loss": 0.0,
742
+ "step": 9200
743
+ },
744
+ {
745
+ "epoch": 4.65,
746
+ "grad_norm": 0.2023557871580124,
747
+ "learning_rate": 5.5406125054816025e-06,
748
+ "loss": 0.0,
749
+ "step": 9300
750
+ },
751
+ {
752
+ "epoch": 4.7,
753
+ "grad_norm": 0.03384646028280258,
754
+ "learning_rate": 5.488835919130032e-06,
755
+ "loss": 0.0,
756
+ "step": 9400
757
+ },
758
+ {
759
+ "epoch": 4.75,
760
+ "grad_norm": 0.04355572536587715,
761
+ "learning_rate": 5.437059332778461e-06,
762
+ "loss": 0.0001,
763
+ "step": 9500
764
+ },
765
+ {
766
+ "epoch": 4.8,
767
+ "grad_norm": 0.02533441036939621,
768
+ "learning_rate": 5.385282746426889e-06,
769
+ "loss": 0.0001,
770
+ "step": 9600
771
+ },
772
+ {
773
+ "epoch": 4.85,
774
+ "grad_norm": 0.10049561411142349,
775
+ "learning_rate": 5.333506160075319e-06,
776
+ "loss": 0.0,
777
+ "step": 9700
778
+ },
779
+ {
780
+ "epoch": 4.9,
781
+ "grad_norm": 0.018155105412006378,
782
+ "learning_rate": 5.281729573723748e-06,
783
+ "loss": 0.0,
784
+ "step": 9800
785
+ },
786
+ {
787
+ "epoch": 4.95,
788
+ "grad_norm": 0.00853881984949112,
789
+ "learning_rate": 5.229952987372177e-06,
790
+ "loss": 0.0,
791
+ "step": 9900
792
+ },
793
+ {
794
+ "epoch": 5.0,
795
+ "grad_norm": 0.17767295241355896,
796
+ "learning_rate": 5.178176401020606e-06,
797
+ "loss": 0.0,
798
+ "step": 10000
799
+ },
800
+ {
801
+ "epoch": 5.0,
802
+ "eval_economic_inequality_accuracy": 0.8261964735516373,
803
+ "eval_economic_inequality_f1": 0.8121087395907841,
804
+ "eval_economic_policy_benefits_accuracy": 0.8653198653198653,
805
+ "eval_economic_policy_benefits_f1": 0.8471939977021603,
806
+ "eval_ethnic_boundaries_accuracy": 0.9362745098039216,
807
+ "eval_ethnic_boundaries_f1": 0.9365673542144131,
808
+ "eval_language_policy_accuracy": 0.7352941176470589,
809
+ "eval_language_policy_f1": 0.6572786690433748,
810
+ "eval_loss": NaN,
811
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
812
+ "eval_mother_tongue_education_f1": 0.837037037037037,
813
+ "eval_overall_accuracy": 0.8553418707750738,
814
+ "eval_overall_f1": 0.8275137305036369,
815
+ "eval_religion_ethnic_identity_accuracy": 0.8800773694390716,
816
+ "eval_religion_ethnic_identity_f1": 0.874896585434052,
817
+ "eval_runtime": 4.4693,
818
+ "eval_samples_per_second": 895.0,
819
+ "eval_steps_per_second": 55.937,
820
+ "step": 10000
821
+ }
822
+ ],
823
+ "logging_steps": 100,
824
+ "max_steps": 20000,
825
+ "num_input_tokens_seen": 0,
826
+ "num_train_epochs": 10,
827
+ "save_steps": 500,
828
+ "stateful_callbacks": {
829
+ "EarlyStoppingCallback": {
830
+ "args": {
831
+ "early_stopping_patience": 2,
832
+ "early_stopping_threshold": 0.0
833
+ },
834
+ "attributes": {
835
+ "early_stopping_patience_counter": 0
836
+ }
837
+ },
838
+ "TrainerControl": {
839
+ "args": {
840
+ "should_epoch_stop": false,
841
+ "should_evaluate": false,
842
+ "should_log": false,
843
+ "should_save": true,
844
+ "should_training_stop": false
845
+ },
846
+ "attributes": {}
847
+ }
848
+ },
849
+ "total_flos": 2.105190825984e+16,
850
+ "train_batch_size": 8,
851
+ "trial_name": null,
852
+ "trial_params": {
853
+ "gradient_accumulation_steps": 1,
854
+ "learning_rate": 1.0001165419669436e-05,
855
+ "num_train_epochs": 10,
856
+ "per_device_train_batch_size": 8,
857
+ "warmup_steps": 684,
858
+ "weight_decay": 0.07522641729243515
859
+ }
860
+ }
run-0/checkpoint-10000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84f9f98f39dc377d902877813dc8cf2a83bc674db97d85b5237e673a48a7375f
3
+ size 5777
run-0/checkpoint-12000/config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3",
16
+ "4": "LABEL_4",
17
+ "5": "LABEL_5",
18
+ "6": "LABEL_6",
19
+ "7": "LABEL_7",
20
+ "8": "LABEL_8",
21
+ "9": "LABEL_9",
22
+ "10": "LABEL_10",
23
+ "11": "LABEL_11",
24
+ "12": "LABEL_12",
25
+ "13": "LABEL_13",
26
+ "14": "LABEL_14",
27
+ "15": "LABEL_15",
28
+ "16": "LABEL_16",
29
+ "17": "LABEL_17"
30
+ },
31
+ "initializer_range": 0.02,
32
+ "intermediate_size": 3072,
33
+ "label2id": {
34
+ "LABEL_0": 0,
35
+ "LABEL_1": 1,
36
+ "LABEL_10": 10,
37
+ "LABEL_11": 11,
38
+ "LABEL_12": 12,
39
+ "LABEL_13": 13,
40
+ "LABEL_14": 14,
41
+ "LABEL_15": 15,
42
+ "LABEL_16": 16,
43
+ "LABEL_17": 17,
44
+ "LABEL_2": 2,
45
+ "LABEL_3": 3,
46
+ "LABEL_4": 4,
47
+ "LABEL_5": 5,
48
+ "LABEL_6": 6,
49
+ "LABEL_7": 7,
50
+ "LABEL_8": 8,
51
+ "LABEL_9": 9
52
+ },
53
+ "layer_norm_eps": 1e-12,
54
+ "max_position_embeddings": 512,
55
+ "model_type": "bert",
56
+ "num_attention_heads": 12,
57
+ "num_hidden_layers": 12,
58
+ "pad_token_id": 0,
59
+ "position_embedding_type": "absolute",
60
+ "torch_dtype": "float32",
61
+ "transformers_version": "4.55.4",
62
+ "type_vocab_size": 2,
63
+ "use_cache": true,
64
+ "vocab_size": 30522
65
+ }
run-0/checkpoint-12000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94a4a57c7f1cb49dfe1fc1b00ee456e5c9e16c12f98099c7ca193f50b949cf3f
3
+ size 438007864
run-0/checkpoint-12000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57efb67745f4bd99aafc8419baf3527b83afde857db0e88e20a0a8c37fdb2ee7
3
+ size 876140043
run-0/checkpoint-12000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2dbc29a654aed70f4abb08c5f8c7a5f58694eaa11f2aefeeb4616e106c8e946
3
+ size 14645
run-0/checkpoint-12000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4345c828504adc5fe4702cd2d2c30ee92faca47578d08c67b12a03686fc460df
3
+ size 1383
run-0/checkpoint-12000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:649a95413e387689b0adfc7b62407f5781fb7473c373f6c6aadd6eafeafd5b3e
3
+ size 1465
run-0/checkpoint-12000/trainer_state.json ADDED
@@ -0,0 +1,1022 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 10000,
3
+ "best_metric": 0.8275137305036369,
4
+ "best_model_checkpoint": "./results/run-0/checkpoint-10000",
5
+ "epoch": 6.0,
6
+ "eval_steps": 500,
7
+ "global_step": 12000,
8
+ "is_hyper_param_search": true,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.05,
14
+ "grad_norm": 2.039576768875122,
15
+ "learning_rate": 1.447537100215313e-06,
16
+ "loss": 0.0,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 0.1,
21
+ "grad_norm": 4.460484504699707,
22
+ "learning_rate": 2.9096957873014878e-06,
23
+ "loss": 0.0821,
24
+ "step": 200
25
+ },
26
+ {
27
+ "epoch": 0.15,
28
+ "grad_norm": 26.84711456298828,
29
+ "learning_rate": 4.371854474387663e-06,
30
+ "loss": 0.0519,
31
+ "step": 300
32
+ },
33
+ {
34
+ "epoch": 0.2,
35
+ "grad_norm": 0.6646966934204102,
36
+ "learning_rate": 5.834013161473838e-06,
37
+ "loss": 0.0,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 0.25,
42
+ "grad_norm": 1.8385223150253296,
43
+ "learning_rate": 7.296171848560012e-06,
44
+ "loss": 0.0,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.3,
49
+ "grad_norm": 0.9234638810157776,
50
+ "learning_rate": 8.758330535646188e-06,
51
+ "loss": 0.0,
52
+ "step": 600
53
+ },
54
+ {
55
+ "epoch": 0.35,
56
+ "grad_norm": 38.24382400512695,
57
+ "learning_rate": 9.9933989317167e-06,
58
+ "loss": 0.0,
59
+ "step": 700
60
+ },
61
+ {
62
+ "epoch": 0.4,
63
+ "grad_norm": 42.83884811401367,
64
+ "learning_rate": 9.94162234536513e-06,
65
+ "loss": 0.0,
66
+ "step": 800
67
+ },
68
+ {
69
+ "epoch": 0.45,
70
+ "grad_norm": 0.8020399808883667,
71
+ "learning_rate": 9.889845759013559e-06,
72
+ "loss": 0.0,
73
+ "step": 900
74
+ },
75
+ {
76
+ "epoch": 0.5,
77
+ "grad_norm": 0.86091148853302,
78
+ "learning_rate": 9.838069172661988e-06,
79
+ "loss": 0.0,
80
+ "step": 1000
81
+ },
82
+ {
83
+ "epoch": 0.55,
84
+ "grad_norm": 0.16439934074878693,
85
+ "learning_rate": 9.786292586310415e-06,
86
+ "loss": 0.0,
87
+ "step": 1100
88
+ },
89
+ {
90
+ "epoch": 0.6,
91
+ "grad_norm": 2.752115488052368,
92
+ "learning_rate": 9.734515999958846e-06,
93
+ "loss": 0.0,
94
+ "step": 1200
95
+ },
96
+ {
97
+ "epoch": 0.65,
98
+ "grad_norm": 1.5777802467346191,
99
+ "learning_rate": 9.682739413607275e-06,
100
+ "loss": 0.0,
101
+ "step": 1300
102
+ },
103
+ {
104
+ "epoch": 0.7,
105
+ "grad_norm": 5.708035945892334,
106
+ "learning_rate": 9.630962827255703e-06,
107
+ "loss": 0.0598,
108
+ "step": 1400
109
+ },
110
+ {
111
+ "epoch": 0.75,
112
+ "grad_norm": 0.3160113990306854,
113
+ "learning_rate": 9.579186240904132e-06,
114
+ "loss": 0.0441,
115
+ "step": 1500
116
+ },
117
+ {
118
+ "epoch": 0.8,
119
+ "grad_norm": 1.2273201942443848,
120
+ "learning_rate": 9.527409654552563e-06,
121
+ "loss": 0.0,
122
+ "step": 1600
123
+ },
124
+ {
125
+ "epoch": 0.85,
126
+ "grad_norm": 29.215551376342773,
127
+ "learning_rate": 9.47563306820099e-06,
128
+ "loss": 0.0,
129
+ "step": 1700
130
+ },
131
+ {
132
+ "epoch": 0.9,
133
+ "grad_norm": 0.18806450068950653,
134
+ "learning_rate": 9.42385648184942e-06,
135
+ "loss": 0.0,
136
+ "step": 1800
137
+ },
138
+ {
139
+ "epoch": 0.95,
140
+ "grad_norm": 0.41723617911338806,
141
+ "learning_rate": 9.372079895497849e-06,
142
+ "loss": 0.0183,
143
+ "step": 1900
144
+ },
145
+ {
146
+ "epoch": 1.0,
147
+ "grad_norm": 1.9550623893737793,
148
+ "learning_rate": 9.32030330914628e-06,
149
+ "loss": 0.0,
150
+ "step": 2000
151
+ },
152
+ {
153
+ "epoch": 1.0,
154
+ "eval_economic_inequality_accuracy": 0.8047858942065491,
155
+ "eval_economic_inequality_f1": 0.7536914695213144,
156
+ "eval_economic_policy_benefits_accuracy": 0.8282828282828283,
157
+ "eval_economic_policy_benefits_f1": 0.7567090529234769,
158
+ "eval_ethnic_boundaries_accuracy": 0.9526143790849673,
159
+ "eval_ethnic_boundaries_f1": 0.9344984933861545,
160
+ "eval_language_policy_accuracy": 0.7058823529411765,
161
+ "eval_language_policy_f1": 0.6071428571428571,
162
+ "eval_loss": NaN,
163
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
164
+ "eval_mother_tongue_education_f1": 0.837037037037037,
165
+ "eval_overall_accuracy": 0.8392311075242039,
166
+ "eval_overall_f1": 0.782435168791527,
167
+ "eval_religion_ethnic_identity_accuracy": 0.8549323017408124,
168
+ "eval_religion_ethnic_identity_f1": 0.8055321027383222,
169
+ "eval_runtime": 4.4874,
170
+ "eval_samples_per_second": 891.378,
171
+ "eval_steps_per_second": 55.711,
172
+ "step": 2000
173
+ },
174
+ {
175
+ "epoch": 1.05,
176
+ "grad_norm": 2.762986183166504,
177
+ "learning_rate": 9.268526722794707e-06,
178
+ "loss": 0.0,
179
+ "step": 2100
180
+ },
181
+ {
182
+ "epoch": 1.1,
183
+ "grad_norm": 0.5487369298934937,
184
+ "learning_rate": 9.216750136443136e-06,
185
+ "loss": 0.0,
186
+ "step": 2200
187
+ },
188
+ {
189
+ "epoch": 1.15,
190
+ "grad_norm": 4.76751184463501,
191
+ "learning_rate": 9.164973550091565e-06,
192
+ "loss": 0.0102,
193
+ "step": 2300
194
+ },
195
+ {
196
+ "epoch": 1.2,
197
+ "grad_norm": 2.681756019592285,
198
+ "learning_rate": 9.113196963739995e-06,
199
+ "loss": 0.0,
200
+ "step": 2400
201
+ },
202
+ {
203
+ "epoch": 1.25,
204
+ "grad_norm": 0.24124185740947723,
205
+ "learning_rate": 9.061420377388424e-06,
206
+ "loss": 0.0,
207
+ "step": 2500
208
+ },
209
+ {
210
+ "epoch": 1.3,
211
+ "grad_norm": 1.943154215812683,
212
+ "learning_rate": 9.009643791036853e-06,
213
+ "loss": 0.0,
214
+ "step": 2600
215
+ },
216
+ {
217
+ "epoch": 1.35,
218
+ "grad_norm": 0.28053101897239685,
219
+ "learning_rate": 8.957867204685282e-06,
220
+ "loss": 0.0347,
221
+ "step": 2700
222
+ },
223
+ {
224
+ "epoch": 1.4,
225
+ "grad_norm": 1.7952568531036377,
226
+ "learning_rate": 8.906090618333711e-06,
227
+ "loss": 0.4436,
228
+ "step": 2800
229
+ },
230
+ {
231
+ "epoch": 1.45,
232
+ "grad_norm": 3.859933614730835,
233
+ "learning_rate": 8.85431403198214e-06,
234
+ "loss": 0.0,
235
+ "step": 2900
236
+ },
237
+ {
238
+ "epoch": 1.5,
239
+ "grad_norm": 1.094464898109436,
240
+ "learning_rate": 8.80253744563057e-06,
241
+ "loss": 0.0,
242
+ "step": 3000
243
+ },
244
+ {
245
+ "epoch": 1.55,
246
+ "grad_norm": 0.5670285820960999,
247
+ "learning_rate": 8.750760859278999e-06,
248
+ "loss": 0.0,
249
+ "step": 3100
250
+ },
251
+ {
252
+ "epoch": 1.6,
253
+ "grad_norm": 2.8412418365478516,
254
+ "learning_rate": 8.698984272927428e-06,
255
+ "loss": 0.0,
256
+ "step": 3200
257
+ },
258
+ {
259
+ "epoch": 1.65,
260
+ "grad_norm": 65.82475280761719,
261
+ "learning_rate": 8.647207686575855e-06,
262
+ "loss": 0.0,
263
+ "step": 3300
264
+ },
265
+ {
266
+ "epoch": 1.7,
267
+ "grad_norm": 8.976615905761719,
268
+ "learning_rate": 8.595431100224286e-06,
269
+ "loss": 0.0048,
270
+ "step": 3400
271
+ },
272
+ {
273
+ "epoch": 1.75,
274
+ "grad_norm": 126.71548461914062,
275
+ "learning_rate": 8.543654513872716e-06,
276
+ "loss": 0.0,
277
+ "step": 3500
278
+ },
279
+ {
280
+ "epoch": 1.8,
281
+ "grad_norm": 0.595015823841095,
282
+ "learning_rate": 8.491877927521143e-06,
283
+ "loss": 0.0033,
284
+ "step": 3600
285
+ },
286
+ {
287
+ "epoch": 1.85,
288
+ "grad_norm": 3.98215651512146,
289
+ "learning_rate": 8.440101341169572e-06,
290
+ "loss": 0.1482,
291
+ "step": 3700
292
+ },
293
+ {
294
+ "epoch": 1.9,
295
+ "grad_norm": 3.0417540073394775,
296
+ "learning_rate": 8.388324754818003e-06,
297
+ "loss": 0.0031,
298
+ "step": 3800
299
+ },
300
+ {
301
+ "epoch": 1.95,
302
+ "grad_norm": 27.85649299621582,
303
+ "learning_rate": 8.33654816846643e-06,
304
+ "loss": 0.0589,
305
+ "step": 3900
306
+ },
307
+ {
308
+ "epoch": 2.0,
309
+ "grad_norm": 2.3094232082366943,
310
+ "learning_rate": 8.28477158211486e-06,
311
+ "loss": 0.0,
312
+ "step": 4000
313
+ },
314
+ {
315
+ "epoch": 2.0,
316
+ "eval_economic_inequality_accuracy": 0.8236775818639799,
317
+ "eval_economic_inequality_f1": 0.8161521311449569,
318
+ "eval_economic_policy_benefits_accuracy": 0.835016835016835,
319
+ "eval_economic_policy_benefits_f1": 0.81117677081242,
320
+ "eval_ethnic_boundaries_accuracy": 0.9477124183006536,
321
+ "eval_ethnic_boundaries_f1": 0.9421262334088456,
322
+ "eval_language_policy_accuracy": 0.7058823529411765,
323
+ "eval_language_policy_f1": 0.6071428571428571,
324
+ "eval_loss": NaN,
325
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
326
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
327
+ "eval_overall_accuracy": 0.8277126438976526,
328
+ "eval_overall_f1": 0.7960173130694123,
329
+ "eval_religion_ethnic_identity_accuracy": 0.8762088974854932,
330
+ "eval_religion_ethnic_identity_f1": 0.8587651451666537,
331
+ "eval_runtime": 4.4654,
332
+ "eval_samples_per_second": 895.775,
333
+ "eval_steps_per_second": 55.986,
334
+ "step": 4000
335
+ },
336
+ {
337
+ "epoch": 2.05,
338
+ "grad_norm": 8.547677040100098,
339
+ "learning_rate": 8.232994995763289e-06,
340
+ "loss": 0.0,
341
+ "step": 4100
342
+ },
343
+ {
344
+ "epoch": 2.1,
345
+ "grad_norm": 117.06266784667969,
346
+ "learning_rate": 8.181218409411718e-06,
347
+ "loss": 0.0,
348
+ "step": 4200
349
+ },
350
+ {
351
+ "epoch": 2.15,
352
+ "grad_norm": 0.19274334609508514,
353
+ "learning_rate": 8.129441823060147e-06,
354
+ "loss": 0.0,
355
+ "step": 4300
356
+ },
357
+ {
358
+ "epoch": 2.2,
359
+ "grad_norm": 1.0360257625579834,
360
+ "learning_rate": 8.077665236708576e-06,
361
+ "loss": 0.0,
362
+ "step": 4400
363
+ },
364
+ {
365
+ "epoch": 2.25,
366
+ "grad_norm": 0.08476172387599945,
367
+ "learning_rate": 8.025888650357006e-06,
368
+ "loss": 0.0,
369
+ "step": 4500
370
+ },
371
+ {
372
+ "epoch": 2.3,
373
+ "grad_norm": 0.08015169203281403,
374
+ "learning_rate": 7.974112064005435e-06,
375
+ "loss": 0.0,
376
+ "step": 4600
377
+ },
378
+ {
379
+ "epoch": 2.35,
380
+ "grad_norm": 0.22844159603118896,
381
+ "learning_rate": 7.922335477653864e-06,
382
+ "loss": 0.0006,
383
+ "step": 4700
384
+ },
385
+ {
386
+ "epoch": 2.4,
387
+ "grad_norm": 0.08015509694814682,
388
+ "learning_rate": 7.870558891302293e-06,
389
+ "loss": 0.0006,
390
+ "step": 4800
391
+ },
392
+ {
393
+ "epoch": 2.45,
394
+ "grad_norm": 5.426948547363281,
395
+ "learning_rate": 7.818782304950722e-06,
396
+ "loss": 0.0,
397
+ "step": 4900
398
+ },
399
+ {
400
+ "epoch": 2.5,
401
+ "grad_norm": 0.022730231285095215,
402
+ "learning_rate": 7.767005718599151e-06,
403
+ "loss": 0.0,
404
+ "step": 5000
405
+ },
406
+ {
407
+ "epoch": 2.55,
408
+ "grad_norm": 5.662137508392334,
409
+ "learning_rate": 7.71522913224758e-06,
410
+ "loss": 0.0,
411
+ "step": 5100
412
+ },
413
+ {
414
+ "epoch": 2.6,
415
+ "grad_norm": 0.05326724052429199,
416
+ "learning_rate": 7.66345254589601e-06,
417
+ "loss": 0.0,
418
+ "step": 5200
419
+ },
420
+ {
421
+ "epoch": 2.65,
422
+ "grad_norm": 2.7219042778015137,
423
+ "learning_rate": 7.611675959544438e-06,
424
+ "loss": 0.0003,
425
+ "step": 5300
426
+ },
427
+ {
428
+ "epoch": 2.7,
429
+ "grad_norm": 4.045983791351318,
430
+ "learning_rate": 7.559899373192868e-06,
431
+ "loss": 0.0,
432
+ "step": 5400
433
+ },
434
+ {
435
+ "epoch": 2.75,
436
+ "grad_norm": 0.17610976099967957,
437
+ "learning_rate": 7.5081227868412964e-06,
438
+ "loss": 0.0,
439
+ "step": 5500
440
+ },
441
+ {
442
+ "epoch": 2.8,
443
+ "grad_norm": 0.03921931982040405,
444
+ "learning_rate": 7.456346200489726e-06,
445
+ "loss": 0.0,
446
+ "step": 5600
447
+ },
448
+ {
449
+ "epoch": 2.85,
450
+ "grad_norm": 3.3493025302886963,
451
+ "learning_rate": 7.404569614138155e-06,
452
+ "loss": 0.0,
453
+ "step": 5700
454
+ },
455
+ {
456
+ "epoch": 2.9,
457
+ "grad_norm": 0.17839455604553223,
458
+ "learning_rate": 7.352793027786584e-06,
459
+ "loss": 0.0,
460
+ "step": 5800
461
+ },
462
+ {
463
+ "epoch": 2.95,
464
+ "grad_norm": 0.01533100288361311,
465
+ "learning_rate": 7.301016441435013e-06,
466
+ "loss": 0.0,
467
+ "step": 5900
468
+ },
469
+ {
470
+ "epoch": 3.0,
471
+ "grad_norm": 0.6168579459190369,
472
+ "learning_rate": 7.249239855083442e-06,
473
+ "loss": 0.0,
474
+ "step": 6000
475
+ },
476
+ {
477
+ "epoch": 3.0,
478
+ "eval_economic_inequality_accuracy": 0.818639798488665,
479
+ "eval_economic_inequality_f1": 0.8128041981528128,
480
+ "eval_economic_policy_benefits_accuracy": 0.8686868686868687,
481
+ "eval_economic_policy_benefits_f1": 0.849679269639744,
482
+ "eval_ethnic_boundaries_accuracy": 0.9379084967320261,
483
+ "eval_ethnic_boundaries_f1": 0.9377319200237987,
484
+ "eval_language_policy_accuracy": 0.6470588235294118,
485
+ "eval_language_policy_f1": 0.5673202614379085,
486
+ "eval_loss": NaN,
487
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
488
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
489
+ "eval_overall_accuracy": 0.8181454231515234,
490
+ "eval_overall_f1": 0.794560338168735,
491
+ "eval_religion_ethnic_identity_accuracy": 0.8588007736943907,
492
+ "eval_religion_ethnic_identity_f1": 0.8590856390174055,
493
+ "eval_runtime": 4.4722,
494
+ "eval_samples_per_second": 894.406,
495
+ "eval_steps_per_second": 55.9,
496
+ "step": 6000
497
+ },
498
+ {
499
+ "epoch": 3.05,
500
+ "grad_norm": 0.06797665357589722,
501
+ "learning_rate": 7.197463268731871e-06,
502
+ "loss": 0.0,
503
+ "step": 6100
504
+ },
505
+ {
506
+ "epoch": 3.1,
507
+ "grad_norm": 0.05142182484269142,
508
+ "learning_rate": 7.145686682380301e-06,
509
+ "loss": 0.0002,
510
+ "step": 6200
511
+ },
512
+ {
513
+ "epoch": 3.15,
514
+ "grad_norm": 0.04868694022297859,
515
+ "learning_rate": 7.09391009602873e-06,
516
+ "loss": 0.0,
517
+ "step": 6300
518
+ },
519
+ {
520
+ "epoch": 3.2,
521
+ "grad_norm": 0.026197949424386024,
522
+ "learning_rate": 7.042133509677158e-06,
523
+ "loss": 0.0005,
524
+ "step": 6400
525
+ },
526
+ {
527
+ "epoch": 3.25,
528
+ "grad_norm": 12.894441604614258,
529
+ "learning_rate": 6.990356923325587e-06,
530
+ "loss": 0.0,
531
+ "step": 6500
532
+ },
533
+ {
534
+ "epoch": 3.3,
535
+ "grad_norm": 4.21354341506958,
536
+ "learning_rate": 6.938580336974017e-06,
537
+ "loss": 0.0,
538
+ "step": 6600
539
+ },
540
+ {
541
+ "epoch": 3.35,
542
+ "grad_norm": 0.0193119365721941,
543
+ "learning_rate": 6.886803750622446e-06,
544
+ "loss": 0.0,
545
+ "step": 6700
546
+ },
547
+ {
548
+ "epoch": 3.4,
549
+ "grad_norm": 0.07511546462774277,
550
+ "learning_rate": 6.835027164270875e-06,
551
+ "loss": 0.0,
552
+ "step": 6800
553
+ },
554
+ {
555
+ "epoch": 3.45,
556
+ "grad_norm": 0.9280033111572266,
557
+ "learning_rate": 6.783250577919304e-06,
558
+ "loss": 0.0003,
559
+ "step": 6900
560
+ },
561
+ {
562
+ "epoch": 3.5,
563
+ "grad_norm": 108.68927001953125,
564
+ "learning_rate": 6.731473991567734e-06,
565
+ "loss": 0.0,
566
+ "step": 7000
567
+ },
568
+ {
569
+ "epoch": 3.55,
570
+ "grad_norm": 4.475680828094482,
571
+ "learning_rate": 6.679697405216162e-06,
572
+ "loss": 0.0,
573
+ "step": 7100
574
+ },
575
+ {
576
+ "epoch": 3.6,
577
+ "grad_norm": 0.040891651064157486,
578
+ "learning_rate": 6.6279208188645915e-06,
579
+ "loss": 0.0,
580
+ "step": 7200
581
+ },
582
+ {
583
+ "epoch": 3.65,
584
+ "grad_norm": 0.032305650413036346,
585
+ "learning_rate": 6.576144232513021e-06,
586
+ "loss": 0.0,
587
+ "step": 7300
588
+ },
589
+ {
590
+ "epoch": 3.7,
591
+ "grad_norm": 0.027986843138933182,
592
+ "learning_rate": 6.524367646161449e-06,
593
+ "loss": 0.0002,
594
+ "step": 7400
595
+ },
596
+ {
597
+ "epoch": 3.75,
598
+ "grad_norm": 0.023173924535512924,
599
+ "learning_rate": 6.472591059809879e-06,
600
+ "loss": 0.0,
601
+ "step": 7500
602
+ },
603
+ {
604
+ "epoch": 3.8,
605
+ "grad_norm": 0.10551729053258896,
606
+ "learning_rate": 6.420814473458308e-06,
607
+ "loss": 0.0,
608
+ "step": 7600
609
+ },
610
+ {
611
+ "epoch": 3.85,
612
+ "grad_norm": 4.631871223449707,
613
+ "learning_rate": 6.3690378871067366e-06,
614
+ "loss": 0.0,
615
+ "step": 7700
616
+ },
617
+ {
618
+ "epoch": 3.9,
619
+ "grad_norm": 0.09392645955085754,
620
+ "learning_rate": 6.317261300755166e-06,
621
+ "loss": 0.0,
622
+ "step": 7800
623
+ },
624
+ {
625
+ "epoch": 3.95,
626
+ "grad_norm": 0.32573947310447693,
627
+ "learning_rate": 6.265484714403596e-06,
628
+ "loss": 0.0,
629
+ "step": 7900
630
+ },
631
+ {
632
+ "epoch": 4.0,
633
+ "grad_norm": 0.012679875828325748,
634
+ "learning_rate": 6.213708128052024e-06,
635
+ "loss": 0.0,
636
+ "step": 8000
637
+ },
638
+ {
639
+ "epoch": 4.0,
640
+ "eval_economic_inequality_accuracy": 0.809823677581864,
641
+ "eval_economic_inequality_f1": 0.8053367188692558,
642
+ "eval_economic_policy_benefits_accuracy": 0.8451178451178452,
643
+ "eval_economic_policy_benefits_f1": 0.8262377279238239,
644
+ "eval_ethnic_boundaries_accuracy": 0.9379084967320261,
645
+ "eval_ethnic_boundaries_f1": 0.9368969810146281,
646
+ "eval_language_policy_accuracy": 0.7058823529411765,
647
+ "eval_language_policy_f1": 0.6071428571428571,
648
+ "eval_loss": NaN,
649
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
650
+ "eval_mother_tongue_education_f1": 0.837037037037037,
651
+ "eval_overall_accuracy": 0.8436493204617509,
652
+ "eval_overall_f1": 0.8135939571355589,
653
+ "eval_religion_ethnic_identity_accuracy": 0.874274661508704,
654
+ "eval_religion_ethnic_identity_f1": 0.8689124208257524,
655
+ "eval_runtime": 4.4538,
656
+ "eval_samples_per_second": 898.1,
657
+ "eval_steps_per_second": 56.131,
658
+ "step": 8000
659
+ },
660
+ {
661
+ "epoch": 4.05,
662
+ "grad_norm": 0.08179322630167007,
663
+ "learning_rate": 6.161931541700453e-06,
664
+ "loss": 0.0,
665
+ "step": 8100
666
+ },
667
+ {
668
+ "epoch": 4.1,
669
+ "grad_norm": 1.8278212547302246,
670
+ "learning_rate": 6.1101549553488824e-06,
671
+ "loss": 0.0,
672
+ "step": 8200
673
+ },
674
+ {
675
+ "epoch": 4.15,
676
+ "grad_norm": 0.9412699341773987,
677
+ "learning_rate": 6.058378368997311e-06,
678
+ "loss": 0.0,
679
+ "step": 8300
680
+ },
681
+ {
682
+ "epoch": 4.2,
683
+ "grad_norm": 0.3116857707500458,
684
+ "learning_rate": 6.006601782645741e-06,
685
+ "loss": 0.0,
686
+ "step": 8400
687
+ },
688
+ {
689
+ "epoch": 4.25,
690
+ "grad_norm": 0.04464244097471237,
691
+ "learning_rate": 5.95482519629417e-06,
692
+ "loss": 0.0,
693
+ "step": 8500
694
+ },
695
+ {
696
+ "epoch": 4.3,
697
+ "grad_norm": 0.0,
698
+ "learning_rate": 5.903048609942598e-06,
699
+ "loss": 0.0,
700
+ "step": 8600
701
+ },
702
+ {
703
+ "epoch": 4.35,
704
+ "grad_norm": 0.011906923726201057,
705
+ "learning_rate": 5.8512720235910274e-06,
706
+ "loss": 0.0,
707
+ "step": 8700
708
+ },
709
+ {
710
+ "epoch": 4.4,
711
+ "grad_norm": 0.022279787808656693,
712
+ "learning_rate": 5.7994954372394575e-06,
713
+ "loss": 0.0002,
714
+ "step": 8800
715
+ },
716
+ {
717
+ "epoch": 4.45,
718
+ "grad_norm": 0.07576213777065277,
719
+ "learning_rate": 5.747718850887886e-06,
720
+ "loss": 0.0001,
721
+ "step": 8900
722
+ },
723
+ {
724
+ "epoch": 4.5,
725
+ "grad_norm": 0.02016213722527027,
726
+ "learning_rate": 5.695942264536315e-06,
727
+ "loss": 0.0,
728
+ "step": 9000
729
+ },
730
+ {
731
+ "epoch": 4.55,
732
+ "grad_norm": 0.07930700480937958,
733
+ "learning_rate": 5.644165678184744e-06,
734
+ "loss": 0.0,
735
+ "step": 9100
736
+ },
737
+ {
738
+ "epoch": 4.6,
739
+ "grad_norm": 0.016734231263399124,
740
+ "learning_rate": 5.592389091833174e-06,
741
+ "loss": 0.0,
742
+ "step": 9200
743
+ },
744
+ {
745
+ "epoch": 4.65,
746
+ "grad_norm": 0.2023557871580124,
747
+ "learning_rate": 5.5406125054816025e-06,
748
+ "loss": 0.0,
749
+ "step": 9300
750
+ },
751
+ {
752
+ "epoch": 4.7,
753
+ "grad_norm": 0.03384646028280258,
754
+ "learning_rate": 5.488835919130032e-06,
755
+ "loss": 0.0,
756
+ "step": 9400
757
+ },
758
+ {
759
+ "epoch": 4.75,
760
+ "grad_norm": 0.04355572536587715,
761
+ "learning_rate": 5.437059332778461e-06,
762
+ "loss": 0.0001,
763
+ "step": 9500
764
+ },
765
+ {
766
+ "epoch": 4.8,
767
+ "grad_norm": 0.02533441036939621,
768
+ "learning_rate": 5.385282746426889e-06,
769
+ "loss": 0.0001,
770
+ "step": 9600
771
+ },
772
+ {
773
+ "epoch": 4.85,
774
+ "grad_norm": 0.10049561411142349,
775
+ "learning_rate": 5.333506160075319e-06,
776
+ "loss": 0.0,
777
+ "step": 9700
778
+ },
779
+ {
780
+ "epoch": 4.9,
781
+ "grad_norm": 0.018155105412006378,
782
+ "learning_rate": 5.281729573723748e-06,
783
+ "loss": 0.0,
784
+ "step": 9800
785
+ },
786
+ {
787
+ "epoch": 4.95,
788
+ "grad_norm": 0.00853881984949112,
789
+ "learning_rate": 5.229952987372177e-06,
790
+ "loss": 0.0,
791
+ "step": 9900
792
+ },
793
+ {
794
+ "epoch": 5.0,
795
+ "grad_norm": 0.17767295241355896,
796
+ "learning_rate": 5.178176401020606e-06,
797
+ "loss": 0.0,
798
+ "step": 10000
799
+ },
800
+ {
801
+ "epoch": 5.0,
802
+ "eval_economic_inequality_accuracy": 0.8261964735516373,
803
+ "eval_economic_inequality_f1": 0.8121087395907841,
804
+ "eval_economic_policy_benefits_accuracy": 0.8653198653198653,
805
+ "eval_economic_policy_benefits_f1": 0.8471939977021603,
806
+ "eval_ethnic_boundaries_accuracy": 0.9362745098039216,
807
+ "eval_ethnic_boundaries_f1": 0.9365673542144131,
808
+ "eval_language_policy_accuracy": 0.7352941176470589,
809
+ "eval_language_policy_f1": 0.6572786690433748,
810
+ "eval_loss": NaN,
811
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
812
+ "eval_mother_tongue_education_f1": 0.837037037037037,
813
+ "eval_overall_accuracy": 0.8553418707750738,
814
+ "eval_overall_f1": 0.8275137305036369,
815
+ "eval_religion_ethnic_identity_accuracy": 0.8800773694390716,
816
+ "eval_religion_ethnic_identity_f1": 0.874896585434052,
817
+ "eval_runtime": 4.4693,
818
+ "eval_samples_per_second": 895.0,
819
+ "eval_steps_per_second": 55.937,
820
+ "step": 10000
821
+ },
822
+ {
823
+ "epoch": 5.05,
824
+ "grad_norm": 0.1333349049091339,
825
+ "learning_rate": 5.126399814669036e-06,
826
+ "loss": 0.0014,
827
+ "step": 10100
828
+ },
829
+ {
830
+ "epoch": 5.1,
831
+ "grad_norm": 1.9743980169296265,
832
+ "learning_rate": 5.074623228317464e-06,
833
+ "loss": 0.0071,
834
+ "step": 10200
835
+ },
836
+ {
837
+ "epoch": 5.15,
838
+ "grad_norm": 0.02137836627662182,
839
+ "learning_rate": 5.022846641965893e-06,
840
+ "loss": 0.0001,
841
+ "step": 10300
842
+ },
843
+ {
844
+ "epoch": 5.2,
845
+ "grad_norm": 28.67204475402832,
846
+ "learning_rate": 4.9710700556143225e-06,
847
+ "loss": 0.0,
848
+ "step": 10400
849
+ },
850
+ {
851
+ "epoch": 5.25,
852
+ "grad_norm": 0.03255374729633331,
853
+ "learning_rate": 4.919293469262752e-06,
854
+ "loss": 0.0011,
855
+ "step": 10500
856
+ },
857
+ {
858
+ "epoch": 5.3,
859
+ "grad_norm": 0.013030334375798702,
860
+ "learning_rate": 4.867516882911181e-06,
861
+ "loss": 0.0,
862
+ "step": 10600
863
+ },
864
+ {
865
+ "epoch": 5.35,
866
+ "grad_norm": 0.006988250184804201,
867
+ "learning_rate": 4.815740296559609e-06,
868
+ "loss": 0.0,
869
+ "step": 10700
870
+ },
871
+ {
872
+ "epoch": 5.4,
873
+ "grad_norm": 0.19381622970104218,
874
+ "learning_rate": 4.763963710208039e-06,
875
+ "loss": 0.0,
876
+ "step": 10800
877
+ },
878
+ {
879
+ "epoch": 5.45,
880
+ "grad_norm": 0.32009562849998474,
881
+ "learning_rate": 4.7121871238564676e-06,
882
+ "loss": 0.0,
883
+ "step": 10900
884
+ },
885
+ {
886
+ "epoch": 5.5,
887
+ "grad_norm": 0.018149327486753464,
888
+ "learning_rate": 4.660410537504898e-06,
889
+ "loss": 0.0,
890
+ "step": 11000
891
+ },
892
+ {
893
+ "epoch": 5.55,
894
+ "grad_norm": 0.02935118041932583,
895
+ "learning_rate": 4.608633951153326e-06,
896
+ "loss": 0.0,
897
+ "step": 11100
898
+ },
899
+ {
900
+ "epoch": 5.6,
901
+ "grad_norm": 0.005307406652718782,
902
+ "learning_rate": 4.556857364801755e-06,
903
+ "loss": 0.0,
904
+ "step": 11200
905
+ },
906
+ {
907
+ "epoch": 5.65,
908
+ "grad_norm": 0.017713243141770363,
909
+ "learning_rate": 4.505080778450184e-06,
910
+ "loss": 0.0,
911
+ "step": 11300
912
+ },
913
+ {
914
+ "epoch": 5.7,
915
+ "grad_norm": 0.4359346330165863,
916
+ "learning_rate": 4.4533041920986134e-06,
917
+ "loss": 0.0,
918
+ "step": 11400
919
+ },
920
+ {
921
+ "epoch": 5.75,
922
+ "grad_norm": 0.007976237684488297,
923
+ "learning_rate": 4.401527605747043e-06,
924
+ "loss": 0.0,
925
+ "step": 11500
926
+ },
927
+ {
928
+ "epoch": 5.8,
929
+ "grad_norm": 0.013159127905964851,
930
+ "learning_rate": 4.349751019395472e-06,
931
+ "loss": 0.0,
932
+ "step": 11600
933
+ },
934
+ {
935
+ "epoch": 5.85,
936
+ "grad_norm": 0.00992471445351839,
937
+ "learning_rate": 4.297974433043901e-06,
938
+ "loss": 0.0,
939
+ "step": 11700
940
+ },
941
+ {
942
+ "epoch": 5.9,
943
+ "grad_norm": 0.030109956860542297,
944
+ "learning_rate": 4.246197846692329e-06,
945
+ "loss": 0.0,
946
+ "step": 11800
947
+ },
948
+ {
949
+ "epoch": 5.95,
950
+ "grad_norm": 1.787908673286438,
951
+ "learning_rate": 4.194421260340759e-06,
952
+ "loss": 0.0,
953
+ "step": 11900
954
+ },
955
+ {
956
+ "epoch": 6.0,
957
+ "grad_norm": 0.026327166706323624,
958
+ "learning_rate": 4.142644673989188e-06,
959
+ "loss": 0.0,
960
+ "step": 12000
961
+ },
962
+ {
963
+ "epoch": 6.0,
964
+ "eval_economic_inequality_accuracy": 0.818639798488665,
965
+ "eval_economic_inequality_f1": 0.8122124279169094,
966
+ "eval_economic_policy_benefits_accuracy": 0.8686868686868687,
967
+ "eval_economic_policy_benefits_f1": 0.8566211399694749,
968
+ "eval_ethnic_boundaries_accuracy": 0.9362745098039216,
969
+ "eval_ethnic_boundaries_f1": 0.9357377620451811,
970
+ "eval_language_policy_accuracy": 0.7352941176470589,
971
+ "eval_language_policy_f1": 0.6586056644880174,
972
+ "eval_loss": NaN,
973
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
974
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
975
+ "eval_overall_accuracy": 0.836769818966157,
976
+ "eval_overall_f1": 0.8138010681888227,
977
+ "eval_religion_ethnic_identity_accuracy": 0.8839458413926499,
978
+ "eval_religion_ethnic_identity_f1": 0.8788886739726133,
979
+ "eval_runtime": 4.4805,
980
+ "eval_samples_per_second": 892.749,
981
+ "eval_steps_per_second": 55.797,
982
+ "step": 12000
983
+ }
984
+ ],
985
+ "logging_steps": 100,
986
+ "max_steps": 20000,
987
+ "num_input_tokens_seen": 0,
988
+ "num_train_epochs": 10,
989
+ "save_steps": 500,
990
+ "stateful_callbacks": {
991
+ "EarlyStoppingCallback": {
992
+ "args": {
993
+ "early_stopping_patience": 2,
994
+ "early_stopping_threshold": 0.0
995
+ },
996
+ "attributes": {
997
+ "early_stopping_patience_counter": 1
998
+ }
999
+ },
1000
+ "TrainerControl": {
1001
+ "args": {
1002
+ "should_epoch_stop": false,
1003
+ "should_evaluate": false,
1004
+ "should_log": false,
1005
+ "should_save": true,
1006
+ "should_training_stop": false
1007
+ },
1008
+ "attributes": {}
1009
+ }
1010
+ },
1011
+ "total_flos": 2.5262289911808e+16,
1012
+ "train_batch_size": 8,
1013
+ "trial_name": null,
1014
+ "trial_params": {
1015
+ "gradient_accumulation_steps": 1,
1016
+ "learning_rate": 1.0001165419669436e-05,
1017
+ "num_train_epochs": 10,
1018
+ "per_device_train_batch_size": 8,
1019
+ "warmup_steps": 684,
1020
+ "weight_decay": 0.07522641729243515
1021
+ }
1022
+ }
run-0/checkpoint-12000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84f9f98f39dc377d902877813dc8cf2a83bc674db97d85b5237e673a48a7375f
3
+ size 5777
run-0/checkpoint-14000/config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3",
16
+ "4": "LABEL_4",
17
+ "5": "LABEL_5",
18
+ "6": "LABEL_6",
19
+ "7": "LABEL_7",
20
+ "8": "LABEL_8",
21
+ "9": "LABEL_9",
22
+ "10": "LABEL_10",
23
+ "11": "LABEL_11",
24
+ "12": "LABEL_12",
25
+ "13": "LABEL_13",
26
+ "14": "LABEL_14",
27
+ "15": "LABEL_15",
28
+ "16": "LABEL_16",
29
+ "17": "LABEL_17"
30
+ },
31
+ "initializer_range": 0.02,
32
+ "intermediate_size": 3072,
33
+ "label2id": {
34
+ "LABEL_0": 0,
35
+ "LABEL_1": 1,
36
+ "LABEL_10": 10,
37
+ "LABEL_11": 11,
38
+ "LABEL_12": 12,
39
+ "LABEL_13": 13,
40
+ "LABEL_14": 14,
41
+ "LABEL_15": 15,
42
+ "LABEL_16": 16,
43
+ "LABEL_17": 17,
44
+ "LABEL_2": 2,
45
+ "LABEL_3": 3,
46
+ "LABEL_4": 4,
47
+ "LABEL_5": 5,
48
+ "LABEL_6": 6,
49
+ "LABEL_7": 7,
50
+ "LABEL_8": 8,
51
+ "LABEL_9": 9
52
+ },
53
+ "layer_norm_eps": 1e-12,
54
+ "max_position_embeddings": 512,
55
+ "model_type": "bert",
56
+ "num_attention_heads": 12,
57
+ "num_hidden_layers": 12,
58
+ "pad_token_id": 0,
59
+ "position_embedding_type": "absolute",
60
+ "torch_dtype": "float32",
61
+ "transformers_version": "4.55.4",
62
+ "type_vocab_size": 2,
63
+ "use_cache": true,
64
+ "vocab_size": 30522
65
+ }
run-0/checkpoint-14000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d54a6abd97fa446a10c97728ae4eacfe0144664b814f0ec4751a71db564b22d4
3
+ size 438007864
run-0/checkpoint-14000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fca96a6b61bfe53a92893af2d1372794f0efee01689fecf47257eacc3b7d9a2
3
+ size 876140043
run-0/checkpoint-14000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df14e1f4e842b9f15d085a26323b8ff19a41f8524bfe152f7179b20f387149ed
3
+ size 14645
run-0/checkpoint-14000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70ccf17f12b51781938dc9a2d2378277e4da0da0109e93e19f7fb3fb9bdeef93
3
+ size 1383
run-0/checkpoint-14000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e9650c04b231ed02de0abdc06c09967974b287f7bc66be11ed787571dfa6863
3
+ size 1465
run-0/checkpoint-14000/trainer_state.json ADDED
@@ -0,0 +1,1184 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 10000,
3
+ "best_metric": 0.8275137305036369,
4
+ "best_model_checkpoint": "./results/run-0/checkpoint-10000",
5
+ "epoch": 7.0,
6
+ "eval_steps": 500,
7
+ "global_step": 14000,
8
+ "is_hyper_param_search": true,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.05,
14
+ "grad_norm": 2.039576768875122,
15
+ "learning_rate": 1.447537100215313e-06,
16
+ "loss": 0.0,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 0.1,
21
+ "grad_norm": 4.460484504699707,
22
+ "learning_rate": 2.9096957873014878e-06,
23
+ "loss": 0.0821,
24
+ "step": 200
25
+ },
26
+ {
27
+ "epoch": 0.15,
28
+ "grad_norm": 26.84711456298828,
29
+ "learning_rate": 4.371854474387663e-06,
30
+ "loss": 0.0519,
31
+ "step": 300
32
+ },
33
+ {
34
+ "epoch": 0.2,
35
+ "grad_norm": 0.6646966934204102,
36
+ "learning_rate": 5.834013161473838e-06,
37
+ "loss": 0.0,
38
+ "step": 400
39
+ },
40
+ {
41
+ "epoch": 0.25,
42
+ "grad_norm": 1.8385223150253296,
43
+ "learning_rate": 7.296171848560012e-06,
44
+ "loss": 0.0,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.3,
49
+ "grad_norm": 0.9234638810157776,
50
+ "learning_rate": 8.758330535646188e-06,
51
+ "loss": 0.0,
52
+ "step": 600
53
+ },
54
+ {
55
+ "epoch": 0.35,
56
+ "grad_norm": 38.24382400512695,
57
+ "learning_rate": 9.9933989317167e-06,
58
+ "loss": 0.0,
59
+ "step": 700
60
+ },
61
+ {
62
+ "epoch": 0.4,
63
+ "grad_norm": 42.83884811401367,
64
+ "learning_rate": 9.94162234536513e-06,
65
+ "loss": 0.0,
66
+ "step": 800
67
+ },
68
+ {
69
+ "epoch": 0.45,
70
+ "grad_norm": 0.8020399808883667,
71
+ "learning_rate": 9.889845759013559e-06,
72
+ "loss": 0.0,
73
+ "step": 900
74
+ },
75
+ {
76
+ "epoch": 0.5,
77
+ "grad_norm": 0.86091148853302,
78
+ "learning_rate": 9.838069172661988e-06,
79
+ "loss": 0.0,
80
+ "step": 1000
81
+ },
82
+ {
83
+ "epoch": 0.55,
84
+ "grad_norm": 0.16439934074878693,
85
+ "learning_rate": 9.786292586310415e-06,
86
+ "loss": 0.0,
87
+ "step": 1100
88
+ },
89
+ {
90
+ "epoch": 0.6,
91
+ "grad_norm": 2.752115488052368,
92
+ "learning_rate": 9.734515999958846e-06,
93
+ "loss": 0.0,
94
+ "step": 1200
95
+ },
96
+ {
97
+ "epoch": 0.65,
98
+ "grad_norm": 1.5777802467346191,
99
+ "learning_rate": 9.682739413607275e-06,
100
+ "loss": 0.0,
101
+ "step": 1300
102
+ },
103
+ {
104
+ "epoch": 0.7,
105
+ "grad_norm": 5.708035945892334,
106
+ "learning_rate": 9.630962827255703e-06,
107
+ "loss": 0.0598,
108
+ "step": 1400
109
+ },
110
+ {
111
+ "epoch": 0.75,
112
+ "grad_norm": 0.3160113990306854,
113
+ "learning_rate": 9.579186240904132e-06,
114
+ "loss": 0.0441,
115
+ "step": 1500
116
+ },
117
+ {
118
+ "epoch": 0.8,
119
+ "grad_norm": 1.2273201942443848,
120
+ "learning_rate": 9.527409654552563e-06,
121
+ "loss": 0.0,
122
+ "step": 1600
123
+ },
124
+ {
125
+ "epoch": 0.85,
126
+ "grad_norm": 29.215551376342773,
127
+ "learning_rate": 9.47563306820099e-06,
128
+ "loss": 0.0,
129
+ "step": 1700
130
+ },
131
+ {
132
+ "epoch": 0.9,
133
+ "grad_norm": 0.18806450068950653,
134
+ "learning_rate": 9.42385648184942e-06,
135
+ "loss": 0.0,
136
+ "step": 1800
137
+ },
138
+ {
139
+ "epoch": 0.95,
140
+ "grad_norm": 0.41723617911338806,
141
+ "learning_rate": 9.372079895497849e-06,
142
+ "loss": 0.0183,
143
+ "step": 1900
144
+ },
145
+ {
146
+ "epoch": 1.0,
147
+ "grad_norm": 1.9550623893737793,
148
+ "learning_rate": 9.32030330914628e-06,
149
+ "loss": 0.0,
150
+ "step": 2000
151
+ },
152
+ {
153
+ "epoch": 1.0,
154
+ "eval_economic_inequality_accuracy": 0.8047858942065491,
155
+ "eval_economic_inequality_f1": 0.7536914695213144,
156
+ "eval_economic_policy_benefits_accuracy": 0.8282828282828283,
157
+ "eval_economic_policy_benefits_f1": 0.7567090529234769,
158
+ "eval_ethnic_boundaries_accuracy": 0.9526143790849673,
159
+ "eval_ethnic_boundaries_f1": 0.9344984933861545,
160
+ "eval_language_policy_accuracy": 0.7058823529411765,
161
+ "eval_language_policy_f1": 0.6071428571428571,
162
+ "eval_loss": NaN,
163
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
164
+ "eval_mother_tongue_education_f1": 0.837037037037037,
165
+ "eval_overall_accuracy": 0.8392311075242039,
166
+ "eval_overall_f1": 0.782435168791527,
167
+ "eval_religion_ethnic_identity_accuracy": 0.8549323017408124,
168
+ "eval_religion_ethnic_identity_f1": 0.8055321027383222,
169
+ "eval_runtime": 4.4874,
170
+ "eval_samples_per_second": 891.378,
171
+ "eval_steps_per_second": 55.711,
172
+ "step": 2000
173
+ },
174
+ {
175
+ "epoch": 1.05,
176
+ "grad_norm": 2.762986183166504,
177
+ "learning_rate": 9.268526722794707e-06,
178
+ "loss": 0.0,
179
+ "step": 2100
180
+ },
181
+ {
182
+ "epoch": 1.1,
183
+ "grad_norm": 0.5487369298934937,
184
+ "learning_rate": 9.216750136443136e-06,
185
+ "loss": 0.0,
186
+ "step": 2200
187
+ },
188
+ {
189
+ "epoch": 1.15,
190
+ "grad_norm": 4.76751184463501,
191
+ "learning_rate": 9.164973550091565e-06,
192
+ "loss": 0.0102,
193
+ "step": 2300
194
+ },
195
+ {
196
+ "epoch": 1.2,
197
+ "grad_norm": 2.681756019592285,
198
+ "learning_rate": 9.113196963739995e-06,
199
+ "loss": 0.0,
200
+ "step": 2400
201
+ },
202
+ {
203
+ "epoch": 1.25,
204
+ "grad_norm": 0.24124185740947723,
205
+ "learning_rate": 9.061420377388424e-06,
206
+ "loss": 0.0,
207
+ "step": 2500
208
+ },
209
+ {
210
+ "epoch": 1.3,
211
+ "grad_norm": 1.943154215812683,
212
+ "learning_rate": 9.009643791036853e-06,
213
+ "loss": 0.0,
214
+ "step": 2600
215
+ },
216
+ {
217
+ "epoch": 1.35,
218
+ "grad_norm": 0.28053101897239685,
219
+ "learning_rate": 8.957867204685282e-06,
220
+ "loss": 0.0347,
221
+ "step": 2700
222
+ },
223
+ {
224
+ "epoch": 1.4,
225
+ "grad_norm": 1.7952568531036377,
226
+ "learning_rate": 8.906090618333711e-06,
227
+ "loss": 0.4436,
228
+ "step": 2800
229
+ },
230
+ {
231
+ "epoch": 1.45,
232
+ "grad_norm": 3.859933614730835,
233
+ "learning_rate": 8.85431403198214e-06,
234
+ "loss": 0.0,
235
+ "step": 2900
236
+ },
237
+ {
238
+ "epoch": 1.5,
239
+ "grad_norm": 1.094464898109436,
240
+ "learning_rate": 8.80253744563057e-06,
241
+ "loss": 0.0,
242
+ "step": 3000
243
+ },
244
+ {
245
+ "epoch": 1.55,
246
+ "grad_norm": 0.5670285820960999,
247
+ "learning_rate": 8.750760859278999e-06,
248
+ "loss": 0.0,
249
+ "step": 3100
250
+ },
251
+ {
252
+ "epoch": 1.6,
253
+ "grad_norm": 2.8412418365478516,
254
+ "learning_rate": 8.698984272927428e-06,
255
+ "loss": 0.0,
256
+ "step": 3200
257
+ },
258
+ {
259
+ "epoch": 1.65,
260
+ "grad_norm": 65.82475280761719,
261
+ "learning_rate": 8.647207686575855e-06,
262
+ "loss": 0.0,
263
+ "step": 3300
264
+ },
265
+ {
266
+ "epoch": 1.7,
267
+ "grad_norm": 8.976615905761719,
268
+ "learning_rate": 8.595431100224286e-06,
269
+ "loss": 0.0048,
270
+ "step": 3400
271
+ },
272
+ {
273
+ "epoch": 1.75,
274
+ "grad_norm": 126.71548461914062,
275
+ "learning_rate": 8.543654513872716e-06,
276
+ "loss": 0.0,
277
+ "step": 3500
278
+ },
279
+ {
280
+ "epoch": 1.8,
281
+ "grad_norm": 0.595015823841095,
282
+ "learning_rate": 8.491877927521143e-06,
283
+ "loss": 0.0033,
284
+ "step": 3600
285
+ },
286
+ {
287
+ "epoch": 1.85,
288
+ "grad_norm": 3.98215651512146,
289
+ "learning_rate": 8.440101341169572e-06,
290
+ "loss": 0.1482,
291
+ "step": 3700
292
+ },
293
+ {
294
+ "epoch": 1.9,
295
+ "grad_norm": 3.0417540073394775,
296
+ "learning_rate": 8.388324754818003e-06,
297
+ "loss": 0.0031,
298
+ "step": 3800
299
+ },
300
+ {
301
+ "epoch": 1.95,
302
+ "grad_norm": 27.85649299621582,
303
+ "learning_rate": 8.33654816846643e-06,
304
+ "loss": 0.0589,
305
+ "step": 3900
306
+ },
307
+ {
308
+ "epoch": 2.0,
309
+ "grad_norm": 2.3094232082366943,
310
+ "learning_rate": 8.28477158211486e-06,
311
+ "loss": 0.0,
312
+ "step": 4000
313
+ },
314
+ {
315
+ "epoch": 2.0,
316
+ "eval_economic_inequality_accuracy": 0.8236775818639799,
317
+ "eval_economic_inequality_f1": 0.8161521311449569,
318
+ "eval_economic_policy_benefits_accuracy": 0.835016835016835,
319
+ "eval_economic_policy_benefits_f1": 0.81117677081242,
320
+ "eval_ethnic_boundaries_accuracy": 0.9477124183006536,
321
+ "eval_ethnic_boundaries_f1": 0.9421262334088456,
322
+ "eval_language_policy_accuracy": 0.7058823529411765,
323
+ "eval_language_policy_f1": 0.6071428571428571,
324
+ "eval_loss": NaN,
325
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
326
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
327
+ "eval_overall_accuracy": 0.8277126438976526,
328
+ "eval_overall_f1": 0.7960173130694123,
329
+ "eval_religion_ethnic_identity_accuracy": 0.8762088974854932,
330
+ "eval_religion_ethnic_identity_f1": 0.8587651451666537,
331
+ "eval_runtime": 4.4654,
332
+ "eval_samples_per_second": 895.775,
333
+ "eval_steps_per_second": 55.986,
334
+ "step": 4000
335
+ },
336
+ {
337
+ "epoch": 2.05,
338
+ "grad_norm": 8.547677040100098,
339
+ "learning_rate": 8.232994995763289e-06,
340
+ "loss": 0.0,
341
+ "step": 4100
342
+ },
343
+ {
344
+ "epoch": 2.1,
345
+ "grad_norm": 117.06266784667969,
346
+ "learning_rate": 8.181218409411718e-06,
347
+ "loss": 0.0,
348
+ "step": 4200
349
+ },
350
+ {
351
+ "epoch": 2.15,
352
+ "grad_norm": 0.19274334609508514,
353
+ "learning_rate": 8.129441823060147e-06,
354
+ "loss": 0.0,
355
+ "step": 4300
356
+ },
357
+ {
358
+ "epoch": 2.2,
359
+ "grad_norm": 1.0360257625579834,
360
+ "learning_rate": 8.077665236708576e-06,
361
+ "loss": 0.0,
362
+ "step": 4400
363
+ },
364
+ {
365
+ "epoch": 2.25,
366
+ "grad_norm": 0.08476172387599945,
367
+ "learning_rate": 8.025888650357006e-06,
368
+ "loss": 0.0,
369
+ "step": 4500
370
+ },
371
+ {
372
+ "epoch": 2.3,
373
+ "grad_norm": 0.08015169203281403,
374
+ "learning_rate": 7.974112064005435e-06,
375
+ "loss": 0.0,
376
+ "step": 4600
377
+ },
378
+ {
379
+ "epoch": 2.35,
380
+ "grad_norm": 0.22844159603118896,
381
+ "learning_rate": 7.922335477653864e-06,
382
+ "loss": 0.0006,
383
+ "step": 4700
384
+ },
385
+ {
386
+ "epoch": 2.4,
387
+ "grad_norm": 0.08015509694814682,
388
+ "learning_rate": 7.870558891302293e-06,
389
+ "loss": 0.0006,
390
+ "step": 4800
391
+ },
392
+ {
393
+ "epoch": 2.45,
394
+ "grad_norm": 5.426948547363281,
395
+ "learning_rate": 7.818782304950722e-06,
396
+ "loss": 0.0,
397
+ "step": 4900
398
+ },
399
+ {
400
+ "epoch": 2.5,
401
+ "grad_norm": 0.022730231285095215,
402
+ "learning_rate": 7.767005718599151e-06,
403
+ "loss": 0.0,
404
+ "step": 5000
405
+ },
406
+ {
407
+ "epoch": 2.55,
408
+ "grad_norm": 5.662137508392334,
409
+ "learning_rate": 7.71522913224758e-06,
410
+ "loss": 0.0,
411
+ "step": 5100
412
+ },
413
+ {
414
+ "epoch": 2.6,
415
+ "grad_norm": 0.05326724052429199,
416
+ "learning_rate": 7.66345254589601e-06,
417
+ "loss": 0.0,
418
+ "step": 5200
419
+ },
420
+ {
421
+ "epoch": 2.65,
422
+ "grad_norm": 2.7219042778015137,
423
+ "learning_rate": 7.611675959544438e-06,
424
+ "loss": 0.0003,
425
+ "step": 5300
426
+ },
427
+ {
428
+ "epoch": 2.7,
429
+ "grad_norm": 4.045983791351318,
430
+ "learning_rate": 7.559899373192868e-06,
431
+ "loss": 0.0,
432
+ "step": 5400
433
+ },
434
+ {
435
+ "epoch": 2.75,
436
+ "grad_norm": 0.17610976099967957,
437
+ "learning_rate": 7.5081227868412964e-06,
438
+ "loss": 0.0,
439
+ "step": 5500
440
+ },
441
+ {
442
+ "epoch": 2.8,
443
+ "grad_norm": 0.03921931982040405,
444
+ "learning_rate": 7.456346200489726e-06,
445
+ "loss": 0.0,
446
+ "step": 5600
447
+ },
448
+ {
449
+ "epoch": 2.85,
450
+ "grad_norm": 3.3493025302886963,
451
+ "learning_rate": 7.404569614138155e-06,
452
+ "loss": 0.0,
453
+ "step": 5700
454
+ },
455
+ {
456
+ "epoch": 2.9,
457
+ "grad_norm": 0.17839455604553223,
458
+ "learning_rate": 7.352793027786584e-06,
459
+ "loss": 0.0,
460
+ "step": 5800
461
+ },
462
+ {
463
+ "epoch": 2.95,
464
+ "grad_norm": 0.01533100288361311,
465
+ "learning_rate": 7.301016441435013e-06,
466
+ "loss": 0.0,
467
+ "step": 5900
468
+ },
469
+ {
470
+ "epoch": 3.0,
471
+ "grad_norm": 0.6168579459190369,
472
+ "learning_rate": 7.249239855083442e-06,
473
+ "loss": 0.0,
474
+ "step": 6000
475
+ },
476
+ {
477
+ "epoch": 3.0,
478
+ "eval_economic_inequality_accuracy": 0.818639798488665,
479
+ "eval_economic_inequality_f1": 0.8128041981528128,
480
+ "eval_economic_policy_benefits_accuracy": 0.8686868686868687,
481
+ "eval_economic_policy_benefits_f1": 0.849679269639744,
482
+ "eval_ethnic_boundaries_accuracy": 0.9379084967320261,
483
+ "eval_ethnic_boundaries_f1": 0.9377319200237987,
484
+ "eval_language_policy_accuracy": 0.6470588235294118,
485
+ "eval_language_policy_f1": 0.5673202614379085,
486
+ "eval_loss": NaN,
487
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
488
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
489
+ "eval_overall_accuracy": 0.8181454231515234,
490
+ "eval_overall_f1": 0.794560338168735,
491
+ "eval_religion_ethnic_identity_accuracy": 0.8588007736943907,
492
+ "eval_religion_ethnic_identity_f1": 0.8590856390174055,
493
+ "eval_runtime": 4.4722,
494
+ "eval_samples_per_second": 894.406,
495
+ "eval_steps_per_second": 55.9,
496
+ "step": 6000
497
+ },
498
+ {
499
+ "epoch": 3.05,
500
+ "grad_norm": 0.06797665357589722,
501
+ "learning_rate": 7.197463268731871e-06,
502
+ "loss": 0.0,
503
+ "step": 6100
504
+ },
505
+ {
506
+ "epoch": 3.1,
507
+ "grad_norm": 0.05142182484269142,
508
+ "learning_rate": 7.145686682380301e-06,
509
+ "loss": 0.0002,
510
+ "step": 6200
511
+ },
512
+ {
513
+ "epoch": 3.15,
514
+ "grad_norm": 0.04868694022297859,
515
+ "learning_rate": 7.09391009602873e-06,
516
+ "loss": 0.0,
517
+ "step": 6300
518
+ },
519
+ {
520
+ "epoch": 3.2,
521
+ "grad_norm": 0.026197949424386024,
522
+ "learning_rate": 7.042133509677158e-06,
523
+ "loss": 0.0005,
524
+ "step": 6400
525
+ },
526
+ {
527
+ "epoch": 3.25,
528
+ "grad_norm": 12.894441604614258,
529
+ "learning_rate": 6.990356923325587e-06,
530
+ "loss": 0.0,
531
+ "step": 6500
532
+ },
533
+ {
534
+ "epoch": 3.3,
535
+ "grad_norm": 4.21354341506958,
536
+ "learning_rate": 6.938580336974017e-06,
537
+ "loss": 0.0,
538
+ "step": 6600
539
+ },
540
+ {
541
+ "epoch": 3.35,
542
+ "grad_norm": 0.0193119365721941,
543
+ "learning_rate": 6.886803750622446e-06,
544
+ "loss": 0.0,
545
+ "step": 6700
546
+ },
547
+ {
548
+ "epoch": 3.4,
549
+ "grad_norm": 0.07511546462774277,
550
+ "learning_rate": 6.835027164270875e-06,
551
+ "loss": 0.0,
552
+ "step": 6800
553
+ },
554
+ {
555
+ "epoch": 3.45,
556
+ "grad_norm": 0.9280033111572266,
557
+ "learning_rate": 6.783250577919304e-06,
558
+ "loss": 0.0003,
559
+ "step": 6900
560
+ },
561
+ {
562
+ "epoch": 3.5,
563
+ "grad_norm": 108.68927001953125,
564
+ "learning_rate": 6.731473991567734e-06,
565
+ "loss": 0.0,
566
+ "step": 7000
567
+ },
568
+ {
569
+ "epoch": 3.55,
570
+ "grad_norm": 4.475680828094482,
571
+ "learning_rate": 6.679697405216162e-06,
572
+ "loss": 0.0,
573
+ "step": 7100
574
+ },
575
+ {
576
+ "epoch": 3.6,
577
+ "grad_norm": 0.040891651064157486,
578
+ "learning_rate": 6.6279208188645915e-06,
579
+ "loss": 0.0,
580
+ "step": 7200
581
+ },
582
+ {
583
+ "epoch": 3.65,
584
+ "grad_norm": 0.032305650413036346,
585
+ "learning_rate": 6.576144232513021e-06,
586
+ "loss": 0.0,
587
+ "step": 7300
588
+ },
589
+ {
590
+ "epoch": 3.7,
591
+ "grad_norm": 0.027986843138933182,
592
+ "learning_rate": 6.524367646161449e-06,
593
+ "loss": 0.0002,
594
+ "step": 7400
595
+ },
596
+ {
597
+ "epoch": 3.75,
598
+ "grad_norm": 0.023173924535512924,
599
+ "learning_rate": 6.472591059809879e-06,
600
+ "loss": 0.0,
601
+ "step": 7500
602
+ },
603
+ {
604
+ "epoch": 3.8,
605
+ "grad_norm": 0.10551729053258896,
606
+ "learning_rate": 6.420814473458308e-06,
607
+ "loss": 0.0,
608
+ "step": 7600
609
+ },
610
+ {
611
+ "epoch": 3.85,
612
+ "grad_norm": 4.631871223449707,
613
+ "learning_rate": 6.3690378871067366e-06,
614
+ "loss": 0.0,
615
+ "step": 7700
616
+ },
617
+ {
618
+ "epoch": 3.9,
619
+ "grad_norm": 0.09392645955085754,
620
+ "learning_rate": 6.317261300755166e-06,
621
+ "loss": 0.0,
622
+ "step": 7800
623
+ },
624
+ {
625
+ "epoch": 3.95,
626
+ "grad_norm": 0.32573947310447693,
627
+ "learning_rate": 6.265484714403596e-06,
628
+ "loss": 0.0,
629
+ "step": 7900
630
+ },
631
+ {
632
+ "epoch": 4.0,
633
+ "grad_norm": 0.012679875828325748,
634
+ "learning_rate": 6.213708128052024e-06,
635
+ "loss": 0.0,
636
+ "step": 8000
637
+ },
638
+ {
639
+ "epoch": 4.0,
640
+ "eval_economic_inequality_accuracy": 0.809823677581864,
641
+ "eval_economic_inequality_f1": 0.8053367188692558,
642
+ "eval_economic_policy_benefits_accuracy": 0.8451178451178452,
643
+ "eval_economic_policy_benefits_f1": 0.8262377279238239,
644
+ "eval_ethnic_boundaries_accuracy": 0.9379084967320261,
645
+ "eval_ethnic_boundaries_f1": 0.9368969810146281,
646
+ "eval_language_policy_accuracy": 0.7058823529411765,
647
+ "eval_language_policy_f1": 0.6071428571428571,
648
+ "eval_loss": NaN,
649
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
650
+ "eval_mother_tongue_education_f1": 0.837037037037037,
651
+ "eval_overall_accuracy": 0.8436493204617509,
652
+ "eval_overall_f1": 0.8135939571355589,
653
+ "eval_religion_ethnic_identity_accuracy": 0.874274661508704,
654
+ "eval_religion_ethnic_identity_f1": 0.8689124208257524,
655
+ "eval_runtime": 4.4538,
656
+ "eval_samples_per_second": 898.1,
657
+ "eval_steps_per_second": 56.131,
658
+ "step": 8000
659
+ },
660
+ {
661
+ "epoch": 4.05,
662
+ "grad_norm": 0.08179322630167007,
663
+ "learning_rate": 6.161931541700453e-06,
664
+ "loss": 0.0,
665
+ "step": 8100
666
+ },
667
+ {
668
+ "epoch": 4.1,
669
+ "grad_norm": 1.8278212547302246,
670
+ "learning_rate": 6.1101549553488824e-06,
671
+ "loss": 0.0,
672
+ "step": 8200
673
+ },
674
+ {
675
+ "epoch": 4.15,
676
+ "grad_norm": 0.9412699341773987,
677
+ "learning_rate": 6.058378368997311e-06,
678
+ "loss": 0.0,
679
+ "step": 8300
680
+ },
681
+ {
682
+ "epoch": 4.2,
683
+ "grad_norm": 0.3116857707500458,
684
+ "learning_rate": 6.006601782645741e-06,
685
+ "loss": 0.0,
686
+ "step": 8400
687
+ },
688
+ {
689
+ "epoch": 4.25,
690
+ "grad_norm": 0.04464244097471237,
691
+ "learning_rate": 5.95482519629417e-06,
692
+ "loss": 0.0,
693
+ "step": 8500
694
+ },
695
+ {
696
+ "epoch": 4.3,
697
+ "grad_norm": 0.0,
698
+ "learning_rate": 5.903048609942598e-06,
699
+ "loss": 0.0,
700
+ "step": 8600
701
+ },
702
+ {
703
+ "epoch": 4.35,
704
+ "grad_norm": 0.011906923726201057,
705
+ "learning_rate": 5.8512720235910274e-06,
706
+ "loss": 0.0,
707
+ "step": 8700
708
+ },
709
+ {
710
+ "epoch": 4.4,
711
+ "grad_norm": 0.022279787808656693,
712
+ "learning_rate": 5.7994954372394575e-06,
713
+ "loss": 0.0002,
714
+ "step": 8800
715
+ },
716
+ {
717
+ "epoch": 4.45,
718
+ "grad_norm": 0.07576213777065277,
719
+ "learning_rate": 5.747718850887886e-06,
720
+ "loss": 0.0001,
721
+ "step": 8900
722
+ },
723
+ {
724
+ "epoch": 4.5,
725
+ "grad_norm": 0.02016213722527027,
726
+ "learning_rate": 5.695942264536315e-06,
727
+ "loss": 0.0,
728
+ "step": 9000
729
+ },
730
+ {
731
+ "epoch": 4.55,
732
+ "grad_norm": 0.07930700480937958,
733
+ "learning_rate": 5.644165678184744e-06,
734
+ "loss": 0.0,
735
+ "step": 9100
736
+ },
737
+ {
738
+ "epoch": 4.6,
739
+ "grad_norm": 0.016734231263399124,
740
+ "learning_rate": 5.592389091833174e-06,
741
+ "loss": 0.0,
742
+ "step": 9200
743
+ },
744
+ {
745
+ "epoch": 4.65,
746
+ "grad_norm": 0.2023557871580124,
747
+ "learning_rate": 5.5406125054816025e-06,
748
+ "loss": 0.0,
749
+ "step": 9300
750
+ },
751
+ {
752
+ "epoch": 4.7,
753
+ "grad_norm": 0.03384646028280258,
754
+ "learning_rate": 5.488835919130032e-06,
755
+ "loss": 0.0,
756
+ "step": 9400
757
+ },
758
+ {
759
+ "epoch": 4.75,
760
+ "grad_norm": 0.04355572536587715,
761
+ "learning_rate": 5.437059332778461e-06,
762
+ "loss": 0.0001,
763
+ "step": 9500
764
+ },
765
+ {
766
+ "epoch": 4.8,
767
+ "grad_norm": 0.02533441036939621,
768
+ "learning_rate": 5.385282746426889e-06,
769
+ "loss": 0.0001,
770
+ "step": 9600
771
+ },
772
+ {
773
+ "epoch": 4.85,
774
+ "grad_norm": 0.10049561411142349,
775
+ "learning_rate": 5.333506160075319e-06,
776
+ "loss": 0.0,
777
+ "step": 9700
778
+ },
779
+ {
780
+ "epoch": 4.9,
781
+ "grad_norm": 0.018155105412006378,
782
+ "learning_rate": 5.281729573723748e-06,
783
+ "loss": 0.0,
784
+ "step": 9800
785
+ },
786
+ {
787
+ "epoch": 4.95,
788
+ "grad_norm": 0.00853881984949112,
789
+ "learning_rate": 5.229952987372177e-06,
790
+ "loss": 0.0,
791
+ "step": 9900
792
+ },
793
+ {
794
+ "epoch": 5.0,
795
+ "grad_norm": 0.17767295241355896,
796
+ "learning_rate": 5.178176401020606e-06,
797
+ "loss": 0.0,
798
+ "step": 10000
799
+ },
800
+ {
801
+ "epoch": 5.0,
802
+ "eval_economic_inequality_accuracy": 0.8261964735516373,
803
+ "eval_economic_inequality_f1": 0.8121087395907841,
804
+ "eval_economic_policy_benefits_accuracy": 0.8653198653198653,
805
+ "eval_economic_policy_benefits_f1": 0.8471939977021603,
806
+ "eval_ethnic_boundaries_accuracy": 0.9362745098039216,
807
+ "eval_ethnic_boundaries_f1": 0.9365673542144131,
808
+ "eval_language_policy_accuracy": 0.7352941176470589,
809
+ "eval_language_policy_f1": 0.6572786690433748,
810
+ "eval_loss": NaN,
811
+ "eval_mother_tongue_education_accuracy": 0.8888888888888888,
812
+ "eval_mother_tongue_education_f1": 0.837037037037037,
813
+ "eval_overall_accuracy": 0.8553418707750738,
814
+ "eval_overall_f1": 0.8275137305036369,
815
+ "eval_religion_ethnic_identity_accuracy": 0.8800773694390716,
816
+ "eval_religion_ethnic_identity_f1": 0.874896585434052,
817
+ "eval_runtime": 4.4693,
818
+ "eval_samples_per_second": 895.0,
819
+ "eval_steps_per_second": 55.937,
820
+ "step": 10000
821
+ },
822
+ {
823
+ "epoch": 5.05,
824
+ "grad_norm": 0.1333349049091339,
825
+ "learning_rate": 5.126399814669036e-06,
826
+ "loss": 0.0014,
827
+ "step": 10100
828
+ },
829
+ {
830
+ "epoch": 5.1,
831
+ "grad_norm": 1.9743980169296265,
832
+ "learning_rate": 5.074623228317464e-06,
833
+ "loss": 0.0071,
834
+ "step": 10200
835
+ },
836
+ {
837
+ "epoch": 5.15,
838
+ "grad_norm": 0.02137836627662182,
839
+ "learning_rate": 5.022846641965893e-06,
840
+ "loss": 0.0001,
841
+ "step": 10300
842
+ },
843
+ {
844
+ "epoch": 5.2,
845
+ "grad_norm": 28.67204475402832,
846
+ "learning_rate": 4.9710700556143225e-06,
847
+ "loss": 0.0,
848
+ "step": 10400
849
+ },
850
+ {
851
+ "epoch": 5.25,
852
+ "grad_norm": 0.03255374729633331,
853
+ "learning_rate": 4.919293469262752e-06,
854
+ "loss": 0.0011,
855
+ "step": 10500
856
+ },
857
+ {
858
+ "epoch": 5.3,
859
+ "grad_norm": 0.013030334375798702,
860
+ "learning_rate": 4.867516882911181e-06,
861
+ "loss": 0.0,
862
+ "step": 10600
863
+ },
864
+ {
865
+ "epoch": 5.35,
866
+ "grad_norm": 0.006988250184804201,
867
+ "learning_rate": 4.815740296559609e-06,
868
+ "loss": 0.0,
869
+ "step": 10700
870
+ },
871
+ {
872
+ "epoch": 5.4,
873
+ "grad_norm": 0.19381622970104218,
874
+ "learning_rate": 4.763963710208039e-06,
875
+ "loss": 0.0,
876
+ "step": 10800
877
+ },
878
+ {
879
+ "epoch": 5.45,
880
+ "grad_norm": 0.32009562849998474,
881
+ "learning_rate": 4.7121871238564676e-06,
882
+ "loss": 0.0,
883
+ "step": 10900
884
+ },
885
+ {
886
+ "epoch": 5.5,
887
+ "grad_norm": 0.018149327486753464,
888
+ "learning_rate": 4.660410537504898e-06,
889
+ "loss": 0.0,
890
+ "step": 11000
891
+ },
892
+ {
893
+ "epoch": 5.55,
894
+ "grad_norm": 0.02935118041932583,
895
+ "learning_rate": 4.608633951153326e-06,
896
+ "loss": 0.0,
897
+ "step": 11100
898
+ },
899
+ {
900
+ "epoch": 5.6,
901
+ "grad_norm": 0.005307406652718782,
902
+ "learning_rate": 4.556857364801755e-06,
903
+ "loss": 0.0,
904
+ "step": 11200
905
+ },
906
+ {
907
+ "epoch": 5.65,
908
+ "grad_norm": 0.017713243141770363,
909
+ "learning_rate": 4.505080778450184e-06,
910
+ "loss": 0.0,
911
+ "step": 11300
912
+ },
913
+ {
914
+ "epoch": 5.7,
915
+ "grad_norm": 0.4359346330165863,
916
+ "learning_rate": 4.4533041920986134e-06,
917
+ "loss": 0.0,
918
+ "step": 11400
919
+ },
920
+ {
921
+ "epoch": 5.75,
922
+ "grad_norm": 0.007976237684488297,
923
+ "learning_rate": 4.401527605747043e-06,
924
+ "loss": 0.0,
925
+ "step": 11500
926
+ },
927
+ {
928
+ "epoch": 5.8,
929
+ "grad_norm": 0.013159127905964851,
930
+ "learning_rate": 4.349751019395472e-06,
931
+ "loss": 0.0,
932
+ "step": 11600
933
+ },
934
+ {
935
+ "epoch": 5.85,
936
+ "grad_norm": 0.00992471445351839,
937
+ "learning_rate": 4.297974433043901e-06,
938
+ "loss": 0.0,
939
+ "step": 11700
940
+ },
941
+ {
942
+ "epoch": 5.9,
943
+ "grad_norm": 0.030109956860542297,
944
+ "learning_rate": 4.246197846692329e-06,
945
+ "loss": 0.0,
946
+ "step": 11800
947
+ },
948
+ {
949
+ "epoch": 5.95,
950
+ "grad_norm": 1.787908673286438,
951
+ "learning_rate": 4.194421260340759e-06,
952
+ "loss": 0.0,
953
+ "step": 11900
954
+ },
955
+ {
956
+ "epoch": 6.0,
957
+ "grad_norm": 0.026327166706323624,
958
+ "learning_rate": 4.142644673989188e-06,
959
+ "loss": 0.0,
960
+ "step": 12000
961
+ },
962
+ {
963
+ "epoch": 6.0,
964
+ "eval_economic_inequality_accuracy": 0.818639798488665,
965
+ "eval_economic_inequality_f1": 0.8122124279169094,
966
+ "eval_economic_policy_benefits_accuracy": 0.8686868686868687,
967
+ "eval_economic_policy_benefits_f1": 0.8566211399694749,
968
+ "eval_ethnic_boundaries_accuracy": 0.9362745098039216,
969
+ "eval_ethnic_boundaries_f1": 0.9357377620451811,
970
+ "eval_language_policy_accuracy": 0.7352941176470589,
971
+ "eval_language_policy_f1": 0.6586056644880174,
972
+ "eval_loss": NaN,
973
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
974
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
975
+ "eval_overall_accuracy": 0.836769818966157,
976
+ "eval_overall_f1": 0.8138010681888227,
977
+ "eval_religion_ethnic_identity_accuracy": 0.8839458413926499,
978
+ "eval_religion_ethnic_identity_f1": 0.8788886739726133,
979
+ "eval_runtime": 4.4805,
980
+ "eval_samples_per_second": 892.749,
981
+ "eval_steps_per_second": 55.797,
982
+ "step": 12000
983
+ },
984
+ {
985
+ "epoch": 6.05,
986
+ "grad_norm": 0.14624720811843872,
987
+ "learning_rate": 4.090868087637617e-06,
988
+ "loss": 0.0,
989
+ "step": 12100
990
+ },
991
+ {
992
+ "epoch": 6.1,
993
+ "grad_norm": 0.035003069788217545,
994
+ "learning_rate": 4.039091501286046e-06,
995
+ "loss": 0.0,
996
+ "step": 12200
997
+ },
998
+ {
999
+ "epoch": 6.15,
1000
+ "grad_norm": 0.006480562500655651,
1001
+ "learning_rate": 3.987314914934475e-06,
1002
+ "loss": 0.0,
1003
+ "step": 12300
1004
+ },
1005
+ {
1006
+ "epoch": 6.2,
1007
+ "grad_norm": 16.347896575927734,
1008
+ "learning_rate": 3.935538328582904e-06,
1009
+ "loss": 0.0,
1010
+ "step": 12400
1011
+ },
1012
+ {
1013
+ "epoch": 6.25,
1014
+ "grad_norm": 0.009049425832927227,
1015
+ "learning_rate": 3.8837617422313335e-06,
1016
+ "loss": 0.0,
1017
+ "step": 12500
1018
+ },
1019
+ {
1020
+ "epoch": 6.3,
1021
+ "grad_norm": 0.16999471187591553,
1022
+ "learning_rate": 3.831985155879763e-06,
1023
+ "loss": 0.0,
1024
+ "step": 12600
1025
+ },
1026
+ {
1027
+ "epoch": 6.35,
1028
+ "grad_norm": 0.010557026602327824,
1029
+ "learning_rate": 3.780208569528192e-06,
1030
+ "loss": 0.0,
1031
+ "step": 12700
1032
+ },
1033
+ {
1034
+ "epoch": 6.4,
1035
+ "grad_norm": 0.46991246938705444,
1036
+ "learning_rate": 3.7284319831766206e-06,
1037
+ "loss": 0.0,
1038
+ "step": 12800
1039
+ },
1040
+ {
1041
+ "epoch": 6.45,
1042
+ "grad_norm": 0.01426410861313343,
1043
+ "learning_rate": 3.6766553968250498e-06,
1044
+ "loss": 0.0,
1045
+ "step": 12900
1046
+ },
1047
+ {
1048
+ "epoch": 6.5,
1049
+ "grad_norm": 0.19090694189071655,
1050
+ "learning_rate": 3.624878810473479e-06,
1051
+ "loss": 0.0001,
1052
+ "step": 13000
1053
+ },
1054
+ {
1055
+ "epoch": 6.55,
1056
+ "grad_norm": 0.014205949380993843,
1057
+ "learning_rate": 3.573102224121908e-06,
1058
+ "loss": 0.2072,
1059
+ "step": 13100
1060
+ },
1061
+ {
1062
+ "epoch": 6.6,
1063
+ "grad_norm": 0.05406644195318222,
1064
+ "learning_rate": 3.521325637770337e-06,
1065
+ "loss": 0.0,
1066
+ "step": 13200
1067
+ },
1068
+ {
1069
+ "epoch": 6.65,
1070
+ "grad_norm": 0.004384028259664774,
1071
+ "learning_rate": 3.4695490514187665e-06,
1072
+ "loss": 0.0,
1073
+ "step": 13300
1074
+ },
1075
+ {
1076
+ "epoch": 6.7,
1077
+ "grad_norm": 0.00611688569188118,
1078
+ "learning_rate": 3.417772465067195e-06,
1079
+ "loss": 0.0,
1080
+ "step": 13400
1081
+ },
1082
+ {
1083
+ "epoch": 6.75,
1084
+ "grad_norm": 0.0035077868960797787,
1085
+ "learning_rate": 3.365995878715625e-06,
1086
+ "loss": 0.0,
1087
+ "step": 13500
1088
+ },
1089
+ {
1090
+ "epoch": 6.8,
1091
+ "grad_norm": 0.013541534543037415,
1092
+ "learning_rate": 3.3142192923640536e-06,
1093
+ "loss": 0.0018,
1094
+ "step": 13600
1095
+ },
1096
+ {
1097
+ "epoch": 6.85,
1098
+ "grad_norm": 0.011575588956475258,
1099
+ "learning_rate": 3.2624427060124823e-06,
1100
+ "loss": 0.0,
1101
+ "step": 13700
1102
+ },
1103
+ {
1104
+ "epoch": 6.9,
1105
+ "grad_norm": 0.012222085148096085,
1106
+ "learning_rate": 3.210666119660912e-06,
1107
+ "loss": 0.0,
1108
+ "step": 13800
1109
+ },
1110
+ {
1111
+ "epoch": 6.95,
1112
+ "grad_norm": 0.18693706393241882,
1113
+ "learning_rate": 3.1588895333093407e-06,
1114
+ "loss": 0.0,
1115
+ "step": 13900
1116
+ },
1117
+ {
1118
+ "epoch": 7.0,
1119
+ "grad_norm": 0.04562217742204666,
1120
+ "learning_rate": 3.10711294695777e-06,
1121
+ "loss": 0.0,
1122
+ "step": 14000
1123
+ },
1124
+ {
1125
+ "epoch": 7.0,
1126
+ "eval_economic_inequality_accuracy": 0.8035264483627204,
1127
+ "eval_economic_inequality_f1": 0.8050953778684486,
1128
+ "eval_economic_policy_benefits_accuracy": 0.8417508417508418,
1129
+ "eval_economic_policy_benefits_f1": 0.8363986338896733,
1130
+ "eval_ethnic_boundaries_accuracy": 0.9183006535947712,
1131
+ "eval_ethnic_boundaries_f1": 0.9256093210513574,
1132
+ "eval_language_policy_accuracy": 0.7058823529411765,
1133
+ "eval_language_policy_f1": 0.6587850826605153,
1134
+ "eval_loss": NaN,
1135
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
1136
+ "eval_mother_tongue_education_f1": 0.7407407407407408,
1137
+ "eval_overall_accuracy": 0.816061278039622,
1138
+ "eval_overall_f1": 0.8032276539443766,
1139
+ "eval_religion_ethnic_identity_accuracy": 0.8491295938104448,
1140
+ "eval_religion_ethnic_identity_f1": 0.8527367674555238,
1141
+ "eval_runtime": 4.4694,
1142
+ "eval_samples_per_second": 894.976,
1143
+ "eval_steps_per_second": 55.936,
1144
+ "step": 14000
1145
+ }
1146
+ ],
1147
+ "logging_steps": 100,
1148
+ "max_steps": 20000,
1149
+ "num_input_tokens_seen": 0,
1150
+ "num_train_epochs": 10,
1151
+ "save_steps": 500,
1152
+ "stateful_callbacks": {
1153
+ "EarlyStoppingCallback": {
1154
+ "args": {
1155
+ "early_stopping_patience": 2,
1156
+ "early_stopping_threshold": 0.0
1157
+ },
1158
+ "attributes": {
1159
+ "early_stopping_patience_counter": 2
1160
+ }
1161
+ },
1162
+ "TrainerControl": {
1163
+ "args": {
1164
+ "should_epoch_stop": false,
1165
+ "should_evaluate": false,
1166
+ "should_log": false,
1167
+ "should_save": true,
1168
+ "should_training_stop": true
1169
+ },
1170
+ "attributes": {}
1171
+ }
1172
+ },
1173
+ "total_flos": 2.9472671563776e+16,
1174
+ "train_batch_size": 8,
1175
+ "trial_name": null,
1176
+ "trial_params": {
1177
+ "gradient_accumulation_steps": 1,
1178
+ "learning_rate": 1.0001165419669436e-05,
1179
+ "num_train_epochs": 10,
1180
+ "per_device_train_batch_size": 8,
1181
+ "warmup_steps": 684,
1182
+ "weight_decay": 0.07522641729243515
1183
+ }
1184
+ }
run-0/checkpoint-14000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84f9f98f39dc377d902877813dc8cf2a83bc674db97d85b5237e673a48a7375f
3
+ size 5777
run-1/checkpoint-125/config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3",
16
+ "4": "LABEL_4",
17
+ "5": "LABEL_5",
18
+ "6": "LABEL_6",
19
+ "7": "LABEL_7",
20
+ "8": "LABEL_8",
21
+ "9": "LABEL_9",
22
+ "10": "LABEL_10",
23
+ "11": "LABEL_11",
24
+ "12": "LABEL_12",
25
+ "13": "LABEL_13",
26
+ "14": "LABEL_14",
27
+ "15": "LABEL_15",
28
+ "16": "LABEL_16",
29
+ "17": "LABEL_17"
30
+ },
31
+ "initializer_range": 0.02,
32
+ "intermediate_size": 3072,
33
+ "label2id": {
34
+ "LABEL_0": 0,
35
+ "LABEL_1": 1,
36
+ "LABEL_10": 10,
37
+ "LABEL_11": 11,
38
+ "LABEL_12": 12,
39
+ "LABEL_13": 13,
40
+ "LABEL_14": 14,
41
+ "LABEL_15": 15,
42
+ "LABEL_16": 16,
43
+ "LABEL_17": 17,
44
+ "LABEL_2": 2,
45
+ "LABEL_3": 3,
46
+ "LABEL_4": 4,
47
+ "LABEL_5": 5,
48
+ "LABEL_6": 6,
49
+ "LABEL_7": 7,
50
+ "LABEL_8": 8,
51
+ "LABEL_9": 9
52
+ },
53
+ "layer_norm_eps": 1e-12,
54
+ "max_position_embeddings": 512,
55
+ "model_type": "bert",
56
+ "num_attention_heads": 12,
57
+ "num_hidden_layers": 12,
58
+ "pad_token_id": 0,
59
+ "position_embedding_type": "absolute",
60
+ "torch_dtype": "float32",
61
+ "transformers_version": "4.55.4",
62
+ "type_vocab_size": 2,
63
+ "use_cache": true,
64
+ "vocab_size": 30522
65
+ }
run-1/checkpoint-125/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52c4ecf0fb267ca0111315770c7aee028de79177925924f83b69d961b456b023
3
+ size 438007864
run-1/checkpoint-125/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e5c78beddc1acbde04851046ecc37f5d997872403e41288893b923d508aa1b9
3
+ size 876140043
run-1/checkpoint-125/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:244592e7bcbeb183331671bf09a9d549c28ce297e22cc46238dcc2c20a3f2316
3
+ size 14645
run-1/checkpoint-125/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3a782a4dbbfc98b2d4cc2a05c2aaf631ec6efdab2dc5da37bf69c5dc9d1ecc7
3
+ size 1383
run-1/checkpoint-125/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:01bcb5fa32e2f77da29ee685d4d5939b9a2b96ff847596b59672a93948497cb2
3
+ size 1465
run-1/checkpoint-125/trainer_state.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 125,
3
+ "best_metric": 0.7222001215424422,
4
+ "best_model_checkpoint": "./results/run-1/checkpoint-125",
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 125,
8
+ "is_hyper_param_search": true,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.8,
14
+ "grad_norm": 12.130753517150879,
15
+ "learning_rate": 1.6523003188214528e-05,
16
+ "loss": 1623.5764,
17
+ "step": 100
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "eval_economic_inequality_accuracy": 0.7770780856423174,
22
+ "eval_economic_inequality_f1": 0.6795991195482776,
23
+ "eval_economic_policy_benefits_accuracy": 0.8215488215488216,
24
+ "eval_economic_policy_benefits_f1": 0.7410643713786043,
25
+ "eval_ethnic_boundaries_accuracy": 0.946078431372549,
26
+ "eval_ethnic_boundaries_f1": 0.9198646713093299,
27
+ "eval_language_policy_accuracy": 0.6764705882352942,
28
+ "eval_language_policy_f1": 0.545923632610939,
29
+ "eval_loss": NaN,
30
+ "eval_mother_tongue_education_accuracy": 0.7777777777777778,
31
+ "eval_mother_tongue_education_f1": 0.6805555555555556,
32
+ "eval_overall_accuracy": 0.8064020197505432,
33
+ "eval_overall_f1": 0.7222001215424422,
34
+ "eval_religion_ethnic_identity_accuracy": 0.839458413926499,
35
+ "eval_religion_ethnic_identity_f1": 0.7661933788519465,
36
+ "eval_runtime": 4.4741,
37
+ "eval_samples_per_second": 894.041,
38
+ "eval_steps_per_second": 55.878,
39
+ "step": 125
40
+ }
41
+ ],
42
+ "logging_steps": 100,
43
+ "max_steps": 2000,
44
+ "num_input_tokens_seen": 0,
45
+ "num_train_epochs": 16,
46
+ "save_steps": 500,
47
+ "stateful_callbacks": {
48
+ "EarlyStoppingCallback": {
49
+ "args": {
50
+ "early_stopping_patience": 2,
51
+ "early_stopping_threshold": 0.0
52
+ },
53
+ "attributes": {
54
+ "early_stopping_patience_counter": 0
55
+ }
56
+ },
57
+ "TrainerControl": {
58
+ "args": {
59
+ "should_epoch_stop": false,
60
+ "should_evaluate": false,
61
+ "should_log": false,
62
+ "should_save": true,
63
+ "should_training_stop": false
64
+ },
65
+ "attributes": {}
66
+ }
67
+ },
68
+ "total_flos": 3368305321574400.0,
69
+ "train_batch_size": 32,
70
+ "trial_name": null,
71
+ "trial_params": {
72
+ "gradient_accumulation_steps": 4,
73
+ "learning_rate": 4.305994770261968e-05,
74
+ "num_train_epochs": 16,
75
+ "per_device_train_batch_size": 32,
76
+ "warmup_steps": 258,
77
+ "weight_decay": 0.09399341365652919
78
+ }
79
+ }
run-1/checkpoint-125/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4410e061e9e9063477d5b57f26eeda29d8b119aba156a0a7c510c00764b1fd9b
3
+ size 5777
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:84f9f98f39dc377d902877813dc8cf2a83bc674db97d85b5237e673a48a7375f
3
  size 5777
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4410e061e9e9063477d5b57f26eeda29d8b119aba156a0a7c510c00764b1fd9b
3
  size 5777