File size: 23,915 Bytes
b5eb89b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
{
  "best_metric": 0.7780054113531673,
  "best_model_checkpoint": "bart-base/checkpoint-10288",
  "epoch": 23.0,
  "eval_steps": 500,
  "global_step": 14789,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.78,
      "learning_rate": 4.871176775531364e-05,
      "loss": 1.2028,
      "step": 500
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.7598760650658405,
      "eval_f1_macro": 0.6063644681337579,
      "eval_f1_micro": 0.7598760650658405,
      "eval_f1_weighted": 0.7489266744747745,
      "eval_loss": 0.8429557681083679,
      "eval_macro_fpr": 0.023226157303183535,
      "eval_macro_sensitivity": 0.6366904609623596,
      "eval_macro_specificity": 0.981694425945952,
      "eval_precision": 0.7600880594489849,
      "eval_precision_macro": 0.6003620126022547,
      "eval_recall": 0.7598760650658405,
      "eval_recall_macro": 0.6366904609623596,
      "eval_runtime": 43.5934,
      "eval_samples_per_second": 29.615,
      "eval_steps_per_second": 3.716,
      "eval_weighted_fpr": 0.02207348333808032,
      "eval_weighted_sensitivity": 0.7598760650658405,
      "eval_weighted_specificity": 0.9655403241234398,
      "step": 643
    },
    {
      "epoch": 1.56,
      "learning_rate": 4.741575946086055e-05,
      "loss": 0.715,
      "step": 1000
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7931835786212239,
      "eval_f1_macro": 0.7213629796675549,
      "eval_f1_micro": 0.793183578621224,
      "eval_f1_weighted": 0.7853477061503886,
      "eval_loss": 0.7331996560096741,
      "eval_macro_fpr": 0.018982515099395962,
      "eval_macro_sensitivity": 0.7320748785168393,
      "eval_macro_specificity": 0.9845140063777027,
      "eval_precision": 0.8019650926098448,
      "eval_precision_macro": 0.7386000214414028,
      "eval_recall": 0.7931835786212239,
      "eval_recall_macro": 0.7320748785168393,
      "eval_runtime": 77.2393,
      "eval_samples_per_second": 16.714,
      "eval_steps_per_second": 2.097,
      "eval_weighted_fpr": 0.018283914264192288,
      "eval_weighted_sensitivity": 0.7931835786212239,
      "eval_weighted_specificity": 0.9745265170443169,
      "step": 1286
    },
    {
      "epoch": 2.33,
      "learning_rate": 4.611975116640747e-05,
      "loss": 0.578,
      "step": 1500
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7939581719597212,
      "eval_f1_macro": 0.6998313050068915,
      "eval_f1_micro": 0.7939581719597211,
      "eval_f1_weighted": 0.79006368267302,
      "eval_loss": 0.8045273423194885,
      "eval_macro_fpr": 0.018545878882531782,
      "eval_macro_sensitivity": 0.7068834234702027,
      "eval_macro_specificity": 0.9847671550097226,
      "eval_precision": 0.8075005963568469,
      "eval_precision_macro": 0.7230632440844518,
      "eval_recall": 0.7939581719597212,
      "eval_recall_macro": 0.7068834234702027,
      "eval_runtime": 86.4099,
      "eval_samples_per_second": 14.94,
      "eval_steps_per_second": 1.875,
      "eval_weighted_fpr": 0.018199233716475097,
      "eval_weighted_sensitivity": 0.7939581719597212,
      "eval_weighted_specificity": 0.977549153186118,
      "step": 1929
    },
    {
      "epoch": 3.11,
      "learning_rate": 4.482633488854329e-05,
      "loss": 0.481,
      "step": 2000
    },
    {
      "epoch": 3.89,
      "learning_rate": 4.3530326594090203e-05,
      "loss": 0.3938,
      "step": 2500
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.8156467854376452,
      "eval_f1_macro": 0.7369315728111729,
      "eval_f1_micro": 0.8156467854376452,
      "eval_f1_weighted": 0.8105168035589256,
      "eval_loss": 0.8291025161743164,
      "eval_macro_fpr": 0.016938074954137138,
      "eval_macro_sensitivity": 0.7217605983180461,
      "eval_macro_specificity": 0.9857842035419169,
      "eval_precision": 0.8171220006778526,
      "eval_precision_macro": 0.7937034642937142,
      "eval_recall": 0.8156467854376452,
      "eval_recall_macro": 0.7217605983180461,
      "eval_runtime": 84.4366,
      "eval_samples_per_second": 15.29,
      "eval_steps_per_second": 1.919,
      "eval_weighted_fpr": 0.01588785046728972,
      "eval_weighted_sensitivity": 0.8156467854376452,
      "eval_weighted_specificity": 0.9711162676911109,
      "step": 2572
    },
    {
      "epoch": 4.67,
      "learning_rate": 4.2236910316226025e-05,
      "loss": 0.3238,
      "step": 3000
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.7939581719597212,
      "eval_f1_macro": 0.7360579650857649,
      "eval_f1_micro": 0.7939581719597211,
      "eval_f1_weighted": 0.7995200606905476,
      "eval_loss": 0.8888681530952454,
      "eval_macro_fpr": 0.018767728188278533,
      "eval_macro_sensitivity": 0.7515467286591376,
      "eval_macro_specificity": 0.9846803211687324,
      "eval_precision": 0.8146370555192632,
      "eval_precision_macro": 0.7464262358644205,
      "eval_recall": 0.7939581719597212,
      "eval_recall_macro": 0.7515467286591376,
      "eval_runtime": 82.179,
      "eval_samples_per_second": 15.71,
      "eval_steps_per_second": 1.971,
      "eval_weighted_fpr": 0.018199233716475097,
      "eval_weighted_sensitivity": 0.7939581719597212,
      "eval_weighted_specificity": 0.9762466455712672,
      "step": 3215
    },
    {
      "epoch": 5.44,
      "learning_rate": 4.094090202177294e-05,
      "loss": 0.246,
      "step": 3500
    },
    {
      "epoch": 6.0,
      "eval_accuracy": 0.7955073586367157,
      "eval_f1_macro": 0.7362210043644567,
      "eval_f1_micro": 0.7955073586367158,
      "eval_f1_weighted": 0.7945977979999089,
      "eval_loss": 1.1628626585006714,
      "eval_macro_fpr": 0.01863187379355594,
      "eval_macro_sensitivity": 0.7599986341032186,
      "eval_macro_specificity": 0.9846932142915057,
      "eval_precision": 0.8066758597712774,
      "eval_precision_macro": 0.7483163388808308,
      "eval_recall": 0.7955073586367157,
      "eval_recall_macro": 0.7599986341032186,
      "eval_runtime": 82.1621,
      "eval_samples_per_second": 15.713,
      "eval_steps_per_second": 1.972,
      "eval_weighted_fpr": 0.018030323726266905,
      "eval_weighted_sensitivity": 0.7955073586367157,
      "eval_weighted_specificity": 0.9748908557358684,
      "step": 3858
    },
    {
      "epoch": 6.22,
      "learning_rate": 3.964489372731986e-05,
      "loss": 0.228,
      "step": 4000
    },
    {
      "epoch": 7.0,
      "learning_rate": 3.834888543286677e-05,
      "loss": 0.1791,
      "step": 4500
    },
    {
      "epoch": 7.0,
      "eval_accuracy": 0.8179705654531371,
      "eval_f1_macro": 0.7736489492761546,
      "eval_f1_micro": 0.817970565453137,
      "eval_f1_weighted": 0.8146592022169725,
      "eval_loss": 1.1354283094406128,
      "eval_macro_fpr": 0.016454731140779742,
      "eval_macro_sensitivity": 0.7696557438486226,
      "eval_macro_specificity": 0.9861772746118799,
      "eval_precision": 0.8150792023791334,
      "eval_precision_macro": 0.7832166669456337,
      "eval_recall": 0.8179705654531371,
      "eval_recall_macro": 0.7696557438486226,
      "eval_runtime": 85.4548,
      "eval_samples_per_second": 15.107,
      "eval_steps_per_second": 1.896,
      "eval_weighted_fpr": 0.01564684732671949,
      "eval_weighted_sensitivity": 0.8179705654531371,
      "eval_weighted_specificity": 0.9746885537250626,
      "step": 4501
    },
    {
      "epoch": 7.78,
      "learning_rate": 3.705546915500259e-05,
      "loss": 0.1305,
      "step": 5000
    },
    {
      "epoch": 8.0,
      "eval_accuracy": 0.8109992254066615,
      "eval_f1_macro": 0.7368975448573482,
      "eval_f1_micro": 0.8109992254066615,
      "eval_f1_weighted": 0.808822910649843,
      "eval_loss": 1.282464861869812,
      "eval_macro_fpr": 0.016943107063520774,
      "eval_macro_sensitivity": 0.7488577570944955,
      "eval_macro_specificity": 0.9858309939135591,
      "eval_precision": 0.8147632227505013,
      "eval_precision_macro": 0.7422132710401315,
      "eval_recall": 0.8109992254066615,
      "eval_recall_macro": 0.7488577570944955,
      "eval_runtime": 87.6049,
      "eval_samples_per_second": 14.737,
      "eval_steps_per_second": 1.849,
      "eval_weighted_fpr": 0.016373641121997046,
      "eval_weighted_sensitivity": 0.8109992254066615,
      "eval_weighted_specificity": 0.9764656832967246,
      "step": 5144
    },
    {
      "epoch": 8.55,
      "learning_rate": 3.575946086054951e-05,
      "loss": 0.0924,
      "step": 5500
    },
    {
      "epoch": 9.0,
      "eval_accuracy": 0.8040278853601859,
      "eval_f1_macro": 0.7559721746710123,
      "eval_f1_micro": 0.8040278853601859,
      "eval_f1_weighted": 0.80145679310201,
      "eval_loss": 1.421708583831787,
      "eval_macro_fpr": 0.017802568540250287,
      "eval_macro_sensitivity": 0.7809236600744539,
      "eval_macro_specificity": 0.9853492039682624,
      "eval_precision": 0.8113877598541575,
      "eval_precision_macro": 0.7464644897074477,
      "eval_recall": 0.8040278853601859,
      "eval_recall_macro": 0.7809236600744539,
      "eval_runtime": 89.5999,
      "eval_samples_per_second": 14.408,
      "eval_steps_per_second": 1.808,
      "eval_weighted_fpr": 0.017111937774771727,
      "eval_weighted_sensitivity": 0.8040278853601859,
      "eval_weighted_specificity": 0.9762101741637506,
      "step": 5787
    },
    {
      "epoch": 9.33,
      "learning_rate": 3.446345256609643e-05,
      "loss": 0.0953,
      "step": 6000
    },
    {
      "epoch": 10.0,
      "eval_accuracy": 0.8024786986831913,
      "eval_f1_macro": 0.7661426626929498,
      "eval_f1_micro": 0.8024786986831914,
      "eval_f1_weighted": 0.8001228885030943,
      "eval_loss": 1.5552372932434082,
      "eval_macro_fpr": 0.018250858387380567,
      "eval_macro_sensitivity": 0.7822186507003115,
      "eval_macro_specificity": 0.9849087494473102,
      "eval_precision": 0.8055922813865947,
      "eval_precision_macro": 0.7702182213697708,
      "eval_recall": 0.8024786986831913,
      "eval_recall_macro": 0.7822186507003115,
      "eval_runtime": 85.2534,
      "eval_samples_per_second": 15.143,
      "eval_steps_per_second": 1.9,
      "eval_weighted_fpr": 0.017277593332881633,
      "eval_weighted_sensitivity": 0.8024786986831913,
      "eval_weighted_specificity": 0.9711525430264618,
      "step": 6430
    },
    {
      "epoch": 10.11,
      "learning_rate": 3.316744427164334e-05,
      "loss": 0.0668,
      "step": 6500
    },
    {
      "epoch": 10.89,
      "learning_rate": 3.1871435977190256e-05,
      "loss": 0.0617,
      "step": 7000
    },
    {
      "epoch": 11.0,
      "eval_accuracy": 0.8040278853601859,
      "eval_f1_macro": 0.7422420279485343,
      "eval_f1_micro": 0.8040278853601859,
      "eval_f1_weighted": 0.8069943234338898,
      "eval_loss": 1.5223894119262695,
      "eval_macro_fpr": 0.017602619829831197,
      "eval_macro_sensitivity": 0.7512050029477755,
      "eval_macro_specificity": 0.9853474796083734,
      "eval_precision": 0.8144253833246468,
      "eval_precision_macro": 0.745702567690635,
      "eval_recall": 0.8040278853601859,
      "eval_recall_macro": 0.7512050029477755,
      "eval_runtime": 85.7004,
      "eval_samples_per_second": 15.064,
      "eval_steps_per_second": 1.89,
      "eval_weighted_fpr": 0.017111937774771727,
      "eval_weighted_sensitivity": 0.8040278853601859,
      "eval_weighted_specificity": 0.9761843087654142,
      "step": 7073
    },
    {
      "epoch": 11.66,
      "learning_rate": 3.057542768273717e-05,
      "loss": 0.0582,
      "step": 7500
    },
    {
      "epoch": 12.0,
      "eval_accuracy": 0.7970565453137103,
      "eval_f1_macro": 0.7465072775190384,
      "eval_f1_micro": 0.7970565453137104,
      "eval_f1_weighted": 0.8021748784206515,
      "eval_loss": 1.6427809000015259,
      "eval_macro_fpr": 0.01832354936918133,
      "eval_macro_sensitivity": 0.7654834054509102,
      "eval_macro_specificity": 0.9849466228972559,
      "eval_precision": 0.8148454054398186,
      "eval_precision_macro": 0.7470312881962948,
      "eval_recall": 0.7970565453137103,
      "eval_recall_macro": 0.7654834054509102,
      "eval_runtime": 95.4565,
      "eval_samples_per_second": 13.524,
      "eval_steps_per_second": 1.697,
      "eval_weighted_fpr": 0.017862012544314153,
      "eval_weighted_sensitivity": 0.7970565453137103,
      "eval_weighted_specificity": 0.9771427981451267,
      "step": 7716
    },
    {
      "epoch": 12.44,
      "learning_rate": 2.9279419388284086e-05,
      "loss": 0.0511,
      "step": 8000
    },
    {
      "epoch": 13.0,
      "eval_accuracy": 0.8195197521301317,
      "eval_f1_macro": 0.7586168211948496,
      "eval_f1_micro": 0.8195197521301317,
      "eval_f1_weighted": 0.8186828900418365,
      "eval_loss": 1.4951962232589722,
      "eval_macro_fpr": 0.016182326988526896,
      "eval_macro_sensitivity": 0.7580273333502953,
      "eval_macro_specificity": 0.9863597979839632,
      "eval_precision": 0.8208277188287908,
      "eval_precision_macro": 0.7645206618886077,
      "eval_recall": 0.8195197521301317,
      "eval_recall_macro": 0.7580273333502953,
      "eval_runtime": 85.8279,
      "eval_samples_per_second": 15.042,
      "eval_steps_per_second": 1.887,
      "eval_weighted_fpr": 0.01548687271518777,
      "eval_weighted_sensitivity": 0.8195197521301317,
      "eval_weighted_specificity": 0.9758772176293167,
      "step": 8359
    },
    {
      "epoch": 13.22,
      "learning_rate": 2.7983411093831004e-05,
      "loss": 0.0403,
      "step": 8500
    },
    {
      "epoch": 14.0,
      "learning_rate": 2.668740279937792e-05,
      "loss": 0.0476,
      "step": 9000
    },
    {
      "epoch": 14.0,
      "eval_accuracy": 0.7970565453137103,
      "eval_f1_macro": 0.7416650578915932,
      "eval_f1_micro": 0.7970565453137104,
      "eval_f1_weighted": 0.7913296065268454,
      "eval_loss": 1.7132102251052856,
      "eval_macro_fpr": 0.018853564171086353,
      "eval_macro_sensitivity": 0.7328360987022614,
      "eval_macro_specificity": 0.9845211676049321,
      "eval_precision": 0.795801987953062,
      "eval_precision_macro": 0.763738385548772,
      "eval_recall": 0.7970565453137103,
      "eval_recall_macro": 0.7328360987022614,
      "eval_runtime": 90.1914,
      "eval_samples_per_second": 14.314,
      "eval_steps_per_second": 1.796,
      "eval_weighted_fpr": 0.017862012544314153,
      "eval_weighted_sensitivity": 0.7970565453137103,
      "eval_weighted_specificity": 0.9707609687602743,
      "step": 9002
    },
    {
      "epoch": 14.77,
      "learning_rate": 2.5391394504924833e-05,
      "loss": 0.0375,
      "step": 9500
    },
    {
      "epoch": 15.0,
      "eval_accuracy": 0.8001549186676995,
      "eval_f1_macro": 0.7437118125777374,
      "eval_f1_micro": 0.8001549186676995,
      "eval_f1_weighted": 0.8017305518379161,
      "eval_loss": 1.7058144807815552,
      "eval_macro_fpr": 0.018188016275518495,
      "eval_macro_sensitivity": 0.7696355635707867,
      "eval_macro_specificity": 0.9850538317438372,
      "eval_precision": 0.8109768674187026,
      "eval_precision_macro": 0.7369403978788467,
      "eval_recall": 0.8001549186676995,
      "eval_recall_macro": 0.7696355635707867,
      "eval_runtime": 89.6171,
      "eval_samples_per_second": 14.406,
      "eval_steps_per_second": 1.808,
      "eval_weighted_fpr": 0.017527173913043478,
      "eval_weighted_sensitivity": 0.8001549186676995,
      "eval_weighted_specificity": 0.9756525574898595,
      "step": 9645
    },
    {
      "epoch": 15.55,
      "learning_rate": 2.4095386210471748e-05,
      "loss": 0.0241,
      "step": 10000
    },
    {
      "epoch": 16.0,
      "eval_accuracy": 0.8179705654531371,
      "eval_f1_macro": 0.7780054113531673,
      "eval_f1_micro": 0.817970565453137,
      "eval_f1_weighted": 0.8174074193030834,
      "eval_loss": 1.7145990133285522,
      "eval_macro_fpr": 0.01624192113353871,
      "eval_macro_sensitivity": 0.7786912615456476,
      "eval_macro_specificity": 0.9862689663514356,
      "eval_precision": 0.8189471437742178,
      "eval_precision_macro": 0.7852139494262297,
      "eval_recall": 0.8179705654531371,
      "eval_recall_macro": 0.7786912615456476,
      "eval_runtime": 95.9969,
      "eval_samples_per_second": 13.448,
      "eval_steps_per_second": 1.688,
      "eval_weighted_fpr": 0.01564684732671949,
      "eval_weighted_sensitivity": 0.8179705654531371,
      "eval_weighted_specificity": 0.9760639298183975,
      "step": 10288
    },
    {
      "epoch": 16.33,
      "learning_rate": 2.280196993260757e-05,
      "loss": 0.0226,
      "step": 10500
    },
    {
      "epoch": 17.0,
      "eval_accuracy": 0.820294345468629,
      "eval_f1_macro": 0.7713653883357572,
      "eval_f1_micro": 0.8202943454686291,
      "eval_f1_weighted": 0.8205584776004088,
      "eval_loss": 1.703484296798706,
      "eval_macro_fpr": 0.015955392684873622,
      "eval_macro_sensitivity": 0.7780791588964692,
      "eval_macro_specificity": 0.9865149979627027,
      "eval_precision": 0.8238197570339643,
      "eval_precision_macro": 0.7731970694747214,
      "eval_recall": 0.820294345468629,
      "eval_recall_macro": 0.7780791588964692,
      "eval_runtime": 92.4633,
      "eval_samples_per_second": 13.962,
      "eval_steps_per_second": 1.752,
      "eval_weighted_fpr": 0.015407092575375215,
      "eval_weighted_sensitivity": 0.820294345468629,
      "eval_weighted_specificity": 0.9774306239719132,
      "step": 10931
    },
    {
      "epoch": 17.11,
      "learning_rate": 2.1505961638154484e-05,
      "loss": 0.017,
      "step": 11000
    },
    {
      "epoch": 17.88,
      "learning_rate": 2.0209953343701402e-05,
      "loss": 0.0189,
      "step": 11500
    },
    {
      "epoch": 18.0,
      "eval_accuracy": 0.8164213787761425,
      "eval_f1_macro": 0.7577737009484876,
      "eval_f1_micro": 0.8164213787761425,
      "eval_f1_weighted": 0.814862576730404,
      "eval_loss": 1.8079380989074707,
      "eval_macro_fpr": 0.016561980735895838,
      "eval_macro_sensitivity": 0.7677111310031827,
      "eval_macro_specificity": 0.9860906809860878,
      "eval_precision": 0.8160431006408752,
      "eval_precision_macro": 0.7583264160007077,
      "eval_recall": 0.8164213787761425,
      "eval_recall_macro": 0.7677111310031827,
      "eval_runtime": 93.2101,
      "eval_samples_per_second": 13.85,
      "eval_steps_per_second": 1.738,
      "eval_weighted_fpr": 0.01580737677582872,
      "eval_weighted_sensitivity": 0.8164213787761425,
      "eval_weighted_specificity": 0.9749388360151747,
      "step": 11574
    },
    {
      "epoch": 18.66,
      "learning_rate": 1.8913945049248317e-05,
      "loss": 0.026,
      "step": 12000
    },
    {
      "epoch": 19.0,
      "eval_accuracy": 0.8125484120836561,
      "eval_f1_macro": 0.7718943136435594,
      "eval_f1_micro": 0.8125484120836561,
      "eval_f1_weighted": 0.8138445092504973,
      "eval_loss": 1.8187031745910645,
      "eval_macro_fpr": 0.016881633170148187,
      "eval_macro_sensitivity": 0.7832768313344278,
      "eval_macro_specificity": 0.9858228106032249,
      "eval_precision": 0.8169875491486152,
      "eval_precision_macro": 0.7674779420581895,
      "eval_recall": 0.8125484120836561,
      "eval_recall_macro": 0.7832768313344278,
      "eval_runtime": 90.0764,
      "eval_samples_per_second": 14.332,
      "eval_steps_per_second": 1.798,
      "eval_weighted_fpr": 0.016211146838156484,
      "eval_weighted_sensitivity": 0.8125484120836561,
      "eval_weighted_specificity": 0.9747937469647154,
      "step": 12217
    },
    {
      "epoch": 19.44,
      "learning_rate": 1.761793675479523e-05,
      "loss": 0.0101,
      "step": 12500
    },
    {
      "epoch": 20.0,
      "eval_accuracy": 0.8187451587916343,
      "eval_f1_macro": 0.7709574718137391,
      "eval_f1_micro": 0.8187451587916342,
      "eval_f1_weighted": 0.8179619842347855,
      "eval_loss": 1.8353675603866577,
      "eval_macro_fpr": 0.01611361621724302,
      "eval_macro_sensitivity": 0.7817701569841005,
      "eval_macro_specificity": 0.9863995310478259,
      "eval_precision": 0.8219549803396689,
      "eval_precision_macro": 0.7748222861859084,
      "eval_recall": 0.8187451587916343,
      "eval_recall_macro": 0.7817701569841005,
      "eval_runtime": 86.3867,
      "eval_samples_per_second": 14.944,
      "eval_steps_per_second": 1.875,
      "eval_weighted_fpr": 0.015566790846194785,
      "eval_weighted_sensitivity": 0.8187451587916343,
      "eval_weighted_specificity": 0.9772478069257525,
      "step": 12860
    },
    {
      "epoch": 20.22,
      "learning_rate": 1.632192846034215e-05,
      "loss": 0.0097,
      "step": 13000
    },
    {
      "epoch": 21.0,
      "learning_rate": 1.502851218247797e-05,
      "loss": 0.0216,
      "step": 13500
    },
    {
      "epoch": 21.0,
      "eval_accuracy": 0.8156467854376452,
      "eval_f1_macro": 0.7618079234508338,
      "eval_f1_micro": 0.8156467854376452,
      "eval_f1_weighted": 0.8164347717885334,
      "eval_loss": 1.8372349739074707,
      "eval_macro_fpr": 0.016334927583449652,
      "eval_macro_sensitivity": 0.7857843546777831,
      "eval_macro_specificity": 0.9863030854921722,
      "eval_precision": 0.8219293193926849,
      "eval_precision_macro": 0.7502422859641604,
      "eval_recall": 0.8156467854376452,
      "eval_recall_macro": 0.7857843546777831,
      "eval_runtime": 87.8538,
      "eval_samples_per_second": 14.695,
      "eval_steps_per_second": 1.844,
      "eval_weighted_fpr": 0.01588785046728972,
      "eval_weighted_sensitivity": 0.8156467854376452,
      "eval_weighted_specificity": 0.9788994969449356,
      "step": 13503
    },
    {
      "epoch": 21.77,
      "learning_rate": 1.3732503888024884e-05,
      "loss": 0.0138,
      "step": 14000
    },
    {
      "epoch": 22.0,
      "eval_accuracy": 0.820294345468629,
      "eval_f1_macro": 0.7661945549993608,
      "eval_f1_micro": 0.8202943454686291,
      "eval_f1_weighted": 0.8222065987286614,
      "eval_loss": 1.8472038507461548,
      "eval_macro_fpr": 0.015872843311496198,
      "eval_macro_sensitivity": 0.7796378584356709,
      "eval_macro_specificity": 0.9865941123818214,
      "eval_precision": 0.8262928188362502,
      "eval_precision_macro": 0.7613141480374475,
      "eval_recall": 0.820294345468629,
      "eval_recall_macro": 0.7796378584356709,
      "eval_runtime": 85.3935,
      "eval_samples_per_second": 15.118,
      "eval_steps_per_second": 1.897,
      "eval_weighted_fpr": 0.015407092575375215,
      "eval_weighted_sensitivity": 0.820294345468629,
      "eval_weighted_specificity": 0.9786173402586936,
      "step": 14146
    },
    {
      "epoch": 22.55,
      "learning_rate": 1.2436495593571799e-05,
      "loss": 0.0169,
      "step": 14500
    },
    {
      "epoch": 23.0,
      "eval_accuracy": 0.8218435321456236,
      "eval_f1_macro": 0.760658140409385,
      "eval_f1_micro": 0.8218435321456236,
      "eval_f1_weighted": 0.8209229699908378,
      "eval_loss": 1.8103588819503784,
      "eval_macro_fpr": 0.016045052484127836,
      "eval_macro_sensitivity": 0.7595040114200765,
      "eval_macro_specificity": 0.986451381533847,
      "eval_precision": 0.8251783607661838,
      "eval_precision_macro": 0.7718974635406236,
      "eval_recall": 0.8218435321456236,
      "eval_recall_macro": 0.7595040114200765,
      "eval_runtime": 92.1092,
      "eval_samples_per_second": 14.016,
      "eval_steps_per_second": 1.759,
      "eval_weighted_fpr": 0.015247944842216919,
      "eval_weighted_sensitivity": 0.8218435321456236,
      "eval_weighted_specificity": 0.9749271908620799,
      "step": 14789
    }
  ],
  "logging_steps": 500,
  "max_steps": 19290,
  "num_train_epochs": 30,
  "save_steps": 500,
  "total_flos": 3.624608859075994e+16,
  "trial_name": null,
  "trial_params": null
}