DunnBC22's picture
End of training
2828de2
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.9963985594237696,
"global_step": 624,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 7.936507936507937e-07,
"loss": 1.0855,
"step": 1
},
{
"epoch": 0.06,
"learning_rate": 9.523809523809523e-06,
"loss": 0.9105,
"step": 12
},
{
"epoch": 0.12,
"learning_rate": 1.9047619047619046e-05,
"loss": 0.4616,
"step": 24
},
{
"epoch": 0.17,
"learning_rate": 2.857142857142857e-05,
"loss": 0.3415,
"step": 36
},
{
"epoch": 0.23,
"learning_rate": 3.809523809523809e-05,
"loss": 0.2655,
"step": 48
},
{
"epoch": 0.29,
"learning_rate": 4.761904761904762e-05,
"loss": 0.265,
"step": 60
},
{
"epoch": 0.35,
"learning_rate": 4.919786096256685e-05,
"loss": 0.2216,
"step": 72
},
{
"epoch": 0.4,
"learning_rate": 4.8128342245989304e-05,
"loss": 0.2317,
"step": 84
},
{
"epoch": 0.46,
"learning_rate": 4.705882352941177e-05,
"loss": 0.194,
"step": 96
},
{
"epoch": 0.52,
"learning_rate": 4.598930481283423e-05,
"loss": 0.1895,
"step": 108
},
{
"epoch": 0.58,
"learning_rate": 4.491978609625669e-05,
"loss": 0.1928,
"step": 120
},
{
"epoch": 0.63,
"learning_rate": 4.385026737967914e-05,
"loss": 0.1852,
"step": 132
},
{
"epoch": 0.69,
"learning_rate": 4.2780748663101606e-05,
"loss": 0.1881,
"step": 144
},
{
"epoch": 0.75,
"learning_rate": 4.171122994652407e-05,
"loss": 0.1763,
"step": 156
},
{
"epoch": 0.81,
"learning_rate": 4.0641711229946525e-05,
"loss": 0.1852,
"step": 168
},
{
"epoch": 0.86,
"learning_rate": 3.957219251336899e-05,
"loss": 0.1757,
"step": 180
},
{
"epoch": 0.92,
"learning_rate": 3.8502673796791445e-05,
"loss": 0.1675,
"step": 192
},
{
"epoch": 0.98,
"learning_rate": 3.743315508021391e-05,
"loss": 0.1535,
"step": 204
},
{
"epoch": 1.0,
"eval_Macro F1": 0.571105133243632,
"eval_Macro Precision": 0.5530940794496672,
"eval_Macro Recall": 0.5924688273721617,
"eval_Micro F1": 0.9622476495882223,
"eval_Micro Precision": 0.9622476495882224,
"eval_Micro Recall": 0.9622476495882224,
"eval_Weighted F1": 0.9596774503435738,
"eval_Weighted Precision": 0.9577464168056835,
"eval_Weighted Recall": 0.9622476495882224,
"eval_accuracy": 0.9622476495882224,
"eval_loss": 0.11255443841218948,
"eval_runtime": 2995.4719,
"eval_samples_per_second": 4.581,
"eval_steps_per_second": 0.143,
"step": 208
},
{
"epoch": 1.04,
"learning_rate": 3.6363636363636364e-05,
"loss": 0.1777,
"step": 216
},
{
"epoch": 1.09,
"learning_rate": 3.529411764705883e-05,
"loss": 0.1577,
"step": 228
},
{
"epoch": 1.15,
"learning_rate": 3.4224598930481284e-05,
"loss": 0.1372,
"step": 240
},
{
"epoch": 1.21,
"learning_rate": 3.3155080213903747e-05,
"loss": 0.1446,
"step": 252
},
{
"epoch": 1.27,
"learning_rate": 3.20855614973262e-05,
"loss": 0.1367,
"step": 264
},
{
"epoch": 1.33,
"learning_rate": 3.1016042780748666e-05,
"loss": 0.1283,
"step": 276
},
{
"epoch": 1.38,
"learning_rate": 2.9946524064171122e-05,
"loss": 0.1521,
"step": 288
},
{
"epoch": 1.44,
"learning_rate": 2.8877005347593582e-05,
"loss": 0.1703,
"step": 300
},
{
"epoch": 1.5,
"learning_rate": 2.7807486631016045e-05,
"loss": 0.1481,
"step": 312
},
{
"epoch": 1.56,
"learning_rate": 2.6737967914438505e-05,
"loss": 0.1353,
"step": 324
},
{
"epoch": 1.61,
"learning_rate": 2.5668449197860968e-05,
"loss": 0.1283,
"step": 336
},
{
"epoch": 1.67,
"learning_rate": 2.4598930481283424e-05,
"loss": 0.1317,
"step": 348
},
{
"epoch": 1.73,
"learning_rate": 2.3529411764705884e-05,
"loss": 0.1344,
"step": 360
},
{
"epoch": 1.79,
"learning_rate": 2.2459893048128343e-05,
"loss": 0.12,
"step": 372
},
{
"epoch": 1.84,
"learning_rate": 2.1390374331550803e-05,
"loss": 0.1232,
"step": 384
},
{
"epoch": 1.9,
"learning_rate": 2.0320855614973263e-05,
"loss": 0.1463,
"step": 396
},
{
"epoch": 1.96,
"learning_rate": 1.9251336898395722e-05,
"loss": 0.1195,
"step": 408
},
{
"epoch": 2.0,
"eval_Macro F1": 0.8501706517182783,
"eval_Macro Precision": 0.9287080582761646,
"eval_Macro Recall": 0.8036955912090624,
"eval_Micro F1": 0.973835726259019,
"eval_Micro Precision": 0.973835726259019,
"eval_Micro Recall": 0.973835726259019,
"eval_Weighted F1": 0.9735756522364626,
"eval_Weighted Precision": 0.9741111567981213,
"eval_Weighted Recall": 0.973835726259019,
"eval_accuracy": 0.973835726259019,
"eval_loss": 0.08434688299894333,
"eval_runtime": 2834.0055,
"eval_samples_per_second": 4.842,
"eval_steps_per_second": 0.151,
"step": 416
},
{
"epoch": 2.02,
"learning_rate": 1.8181818181818182e-05,
"loss": 0.1102,
"step": 420
},
{
"epoch": 2.07,
"learning_rate": 1.7112299465240642e-05,
"loss": 0.1248,
"step": 432
},
{
"epoch": 2.13,
"learning_rate": 1.60427807486631e-05,
"loss": 0.108,
"step": 444
},
{
"epoch": 2.19,
"learning_rate": 1.4973262032085561e-05,
"loss": 0.118,
"step": 456
},
{
"epoch": 2.25,
"learning_rate": 1.3903743315508022e-05,
"loss": 0.1131,
"step": 468
},
{
"epoch": 2.3,
"learning_rate": 1.2834224598930484e-05,
"loss": 0.1177,
"step": 480
},
{
"epoch": 2.36,
"learning_rate": 1.1764705882352942e-05,
"loss": 0.1409,
"step": 492
},
{
"epoch": 2.42,
"learning_rate": 1.0695187165775402e-05,
"loss": 0.1261,
"step": 504
},
{
"epoch": 2.48,
"learning_rate": 9.625668449197861e-06,
"loss": 0.1004,
"step": 516
},
{
"epoch": 2.54,
"learning_rate": 8.556149732620321e-06,
"loss": 0.1269,
"step": 528
},
{
"epoch": 2.59,
"learning_rate": 7.4866310160427806e-06,
"loss": 0.1265,
"step": 540
},
{
"epoch": 2.65,
"learning_rate": 6.417112299465242e-06,
"loss": 0.098,
"step": 552
},
{
"epoch": 2.71,
"learning_rate": 5.347593582887701e-06,
"loss": 0.126,
"step": 564
},
{
"epoch": 2.77,
"learning_rate": 4.2780748663101604e-06,
"loss": 0.0876,
"step": 576
},
{
"epoch": 2.82,
"learning_rate": 3.208556149732621e-06,
"loss": 0.1014,
"step": 588
},
{
"epoch": 2.88,
"learning_rate": 2.1390374331550802e-06,
"loss": 0.1111,
"step": 600
},
{
"epoch": 2.94,
"learning_rate": 1.0695187165775401e-06,
"loss": 0.1361,
"step": 612
},
{
"epoch": 3.0,
"learning_rate": 0.0,
"loss": 0.0979,
"step": 624
},
{
"epoch": 3.0,
"eval_Macro F1": 0.9154020288755031,
"eval_Macro Precision": 0.9314436651998532,
"eval_Macro Recall": 0.9019139504496776,
"eval_Micro F1": 0.976678084687705,
"eval_Micro Precision": 0.976678084687705,
"eval_Micro Recall": 0.976678084687705,
"eval_Weighted F1": 0.9768442651021538,
"eval_Weighted Precision": 0.9771017843393915,
"eval_Weighted Recall": 0.976678084687705,
"eval_accuracy": 0.976678084687705,
"eval_loss": 0.07859531044960022,
"eval_runtime": 3028.7856,
"eval_samples_per_second": 4.53,
"eval_steps_per_second": 0.142,
"step": 624
},
{
"epoch": 3.0,
"step": 624,
"total_flos": 6.186805817770303e+18,
"train_loss": 0.17273353269466987,
"train_runtime": 60970.1196,
"train_samples_per_second": 1.311,
"train_steps_per_second": 0.01
}
],
"max_steps": 624,
"num_train_epochs": 3,
"total_flos": 6.186805817770303e+18,
"trial_name": null,
"trial_params": null
}