sbf-bart-sex / checkpoint-1110 /trainer_state.json
anoaky's picture
Training in progress, epoch 2, checkpoint
4897c7a verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 693,
"global_step": 1110,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0,
"eval_f1": 0.14772727272727273,
"eval_loss": 0.8850772380828857,
"eval_precision": 0.08080808080808081,
"eval_recall": 0.859504132231405,
"eval_runtime": 99.3343,
"eval_samples_per_second": 47.043,
"eval_steps_per_second": 5.889,
"step": 0
},
{
"epoch": 0.018026137899954935,
"grad_norm": 6.000438690185547,
"learning_rate": 4.981949458483755e-05,
"loss": 0.3931,
"step": 10
},
{
"epoch": 0.03605227579990987,
"grad_norm": 47.497467041015625,
"learning_rate": 4.963898916967509e-05,
"loss": 0.2607,
"step": 20
},
{
"epoch": 0.054078413699864804,
"grad_norm": 13.464141845703125,
"learning_rate": 4.945848375451264e-05,
"loss": 0.1813,
"step": 30
},
{
"epoch": 0.07210455159981974,
"grad_norm": 15.57222843170166,
"learning_rate": 4.927797833935018e-05,
"loss": 0.2618,
"step": 40
},
{
"epoch": 0.09013068949977468,
"grad_norm": 11.892234802246094,
"learning_rate": 4.909747292418773e-05,
"loss": 0.1559,
"step": 50
},
{
"epoch": 0.10815682739972961,
"grad_norm": 6.340595722198486,
"learning_rate": 4.891696750902527e-05,
"loss": 0.1648,
"step": 60
},
{
"epoch": 0.12618296529968454,
"grad_norm": 10.196683883666992,
"learning_rate": 4.873646209386282e-05,
"loss": 0.1472,
"step": 70
},
{
"epoch": 0.14420910319963948,
"grad_norm": 11.511515617370605,
"learning_rate": 4.855595667870036e-05,
"loss": 0.2108,
"step": 80
},
{
"epoch": 0.16223524109959442,
"grad_norm": 8.681073188781738,
"learning_rate": 4.837545126353791e-05,
"loss": 0.1561,
"step": 90
},
{
"epoch": 0.18026137899954936,
"grad_norm": 41.062801361083984,
"learning_rate": 4.819494584837546e-05,
"loss": 0.1563,
"step": 100
},
{
"epoch": 0.19828751689950427,
"grad_norm": 19.93891716003418,
"learning_rate": 4.8014440433213e-05,
"loss": 0.1881,
"step": 110
},
{
"epoch": 0.21631365479945922,
"grad_norm": 21.90877914428711,
"learning_rate": 4.783393501805055e-05,
"loss": 0.1262,
"step": 120
},
{
"epoch": 0.23433979269941416,
"grad_norm": 14.123740196228027,
"learning_rate": 4.765342960288809e-05,
"loss": 0.1535,
"step": 130
},
{
"epoch": 0.25236593059936907,
"grad_norm": 30.671157836914062,
"learning_rate": 4.747292418772563e-05,
"loss": 0.1458,
"step": 140
},
{
"epoch": 0.270392068499324,
"grad_norm": 7.560745716094971,
"learning_rate": 4.7292418772563177e-05,
"loss": 0.1797,
"step": 150
},
{
"epoch": 0.28841820639927895,
"grad_norm": 12.448927879333496,
"learning_rate": 4.711191335740072e-05,
"loss": 0.1898,
"step": 160
},
{
"epoch": 0.3064443442992339,
"grad_norm": 20.040008544921875,
"learning_rate": 4.693140794223827e-05,
"loss": 0.1698,
"step": 170
},
{
"epoch": 0.32447048219918884,
"grad_norm": 7.592446804046631,
"learning_rate": 4.675090252707581e-05,
"loss": 0.156,
"step": 180
},
{
"epoch": 0.3424966200991438,
"grad_norm": 18.830904006958008,
"learning_rate": 4.657039711191336e-05,
"loss": 0.1536,
"step": 190
},
{
"epoch": 0.3605227579990987,
"grad_norm": 3.684938430786133,
"learning_rate": 4.63898916967509e-05,
"loss": 0.1152,
"step": 200
},
{
"epoch": 0.3785488958990536,
"grad_norm": 13.293309211730957,
"learning_rate": 4.620938628158845e-05,
"loss": 0.2116,
"step": 210
},
{
"epoch": 0.39657503379900855,
"grad_norm": 14.630786895751953,
"learning_rate": 4.602888086642599e-05,
"loss": 0.1549,
"step": 220
},
{
"epoch": 0.4146011716989635,
"grad_norm": 20.37642478942871,
"learning_rate": 4.584837545126354e-05,
"loss": 0.1679,
"step": 230
},
{
"epoch": 0.43262730959891843,
"grad_norm": 13.183340072631836,
"learning_rate": 4.566787003610109e-05,
"loss": 0.1581,
"step": 240
},
{
"epoch": 0.45065344749887337,
"grad_norm": 21.777263641357422,
"learning_rate": 4.548736462093863e-05,
"loss": 0.1618,
"step": 250
},
{
"epoch": 0.4686795853988283,
"grad_norm": 20.683944702148438,
"learning_rate": 4.530685920577618e-05,
"loss": 0.1606,
"step": 260
},
{
"epoch": 0.48670572329878325,
"grad_norm": 30.506752014160156,
"learning_rate": 4.5126353790613716e-05,
"loss": 0.1831,
"step": 270
},
{
"epoch": 0.5047318611987381,
"grad_norm": 22.953815460205078,
"learning_rate": 4.494584837545127e-05,
"loss": 0.1565,
"step": 280
},
{
"epoch": 0.5227579990986931,
"grad_norm": 6.177365303039551,
"learning_rate": 4.4765342960288806e-05,
"loss": 0.1287,
"step": 290
},
{
"epoch": 0.540784136998648,
"grad_norm": 8.373098373413086,
"learning_rate": 4.458483754512636e-05,
"loss": 0.1553,
"step": 300
},
{
"epoch": 0.558810274898603,
"grad_norm": 10.07934284210205,
"learning_rate": 4.44043321299639e-05,
"loss": 0.1868,
"step": 310
},
{
"epoch": 0.5768364127985579,
"grad_norm": 18.686702728271484,
"learning_rate": 4.422382671480145e-05,
"loss": 0.1576,
"step": 320
},
{
"epoch": 0.5948625506985128,
"grad_norm": 10.679413795471191,
"learning_rate": 4.404332129963899e-05,
"loss": 0.1409,
"step": 330
},
{
"epoch": 0.6128886885984678,
"grad_norm": 11.381563186645508,
"learning_rate": 4.386281588447654e-05,
"loss": 0.2043,
"step": 340
},
{
"epoch": 0.6309148264984227,
"grad_norm": 40.68292236328125,
"learning_rate": 4.368231046931408e-05,
"loss": 0.1288,
"step": 350
},
{
"epoch": 0.6489409643983777,
"grad_norm": 17.46027946472168,
"learning_rate": 4.350180505415163e-05,
"loss": 0.1537,
"step": 360
},
{
"epoch": 0.6669671022983326,
"grad_norm": 13.656320571899414,
"learning_rate": 4.332129963898917e-05,
"loss": 0.1201,
"step": 370
},
{
"epoch": 0.6849932401982876,
"grad_norm": 20.412939071655273,
"learning_rate": 4.314079422382672e-05,
"loss": 0.1321,
"step": 380
},
{
"epoch": 0.7030193780982424,
"grad_norm": 9.957433700561523,
"learning_rate": 4.296028880866426e-05,
"loss": 0.141,
"step": 390
},
{
"epoch": 0.7210455159981974,
"grad_norm": 9.682558059692383,
"learning_rate": 4.277978339350181e-05,
"loss": 0.1607,
"step": 400
},
{
"epoch": 0.7390716538981523,
"grad_norm": 9.733131408691406,
"learning_rate": 4.259927797833935e-05,
"loss": 0.1037,
"step": 410
},
{
"epoch": 0.7570977917981072,
"grad_norm": 15.76230525970459,
"learning_rate": 4.24187725631769e-05,
"loss": 0.2101,
"step": 420
},
{
"epoch": 0.7751239296980622,
"grad_norm": 22.166545867919922,
"learning_rate": 4.223826714801444e-05,
"loss": 0.1459,
"step": 430
},
{
"epoch": 0.7931500675980171,
"grad_norm": 17.87278175354004,
"learning_rate": 4.205776173285199e-05,
"loss": 0.147,
"step": 440
},
{
"epoch": 0.8111762054979721,
"grad_norm": 5.201169490814209,
"learning_rate": 4.187725631768953e-05,
"loss": 0.1247,
"step": 450
},
{
"epoch": 0.829202343397927,
"grad_norm": 8.68251895904541,
"learning_rate": 4.169675090252708e-05,
"loss": 0.1172,
"step": 460
},
{
"epoch": 0.847228481297882,
"grad_norm": 14.086528778076172,
"learning_rate": 4.151624548736462e-05,
"loss": 0.111,
"step": 470
},
{
"epoch": 0.8652546191978369,
"grad_norm": 5.421584606170654,
"learning_rate": 4.1335740072202167e-05,
"loss": 0.1264,
"step": 480
},
{
"epoch": 0.8832807570977917,
"grad_norm": 7.700821399688721,
"learning_rate": 4.115523465703972e-05,
"loss": 0.1278,
"step": 490
},
{
"epoch": 0.9013068949977467,
"grad_norm": 9.820999145507812,
"learning_rate": 4.0974729241877256e-05,
"loss": 0.1294,
"step": 500
},
{
"epoch": 0.9193330328977016,
"grad_norm": 15.016349792480469,
"learning_rate": 4.079422382671481e-05,
"loss": 0.1607,
"step": 510
},
{
"epoch": 0.9373591707976566,
"grad_norm": 10.43041706085205,
"learning_rate": 4.0613718411552346e-05,
"loss": 0.1091,
"step": 520
},
{
"epoch": 0.9553853086976115,
"grad_norm": 26.99540901184082,
"learning_rate": 4.043321299638989e-05,
"loss": 0.1442,
"step": 530
},
{
"epoch": 0.9734114465975665,
"grad_norm": 8.318906784057617,
"learning_rate": 4.0252707581227436e-05,
"loss": 0.1233,
"step": 540
},
{
"epoch": 0.9914375844975214,
"grad_norm": 8.85923957824707,
"learning_rate": 4.007220216606498e-05,
"loss": 0.1466,
"step": 550
},
{
"epoch": 1.0090130689499774,
"grad_norm": 4.69356107711792,
"learning_rate": 3.989169675090253e-05,
"loss": 0.1433,
"step": 560
},
{
"epoch": 1.0270392068499323,
"grad_norm": 17.66107177734375,
"learning_rate": 3.971119133574007e-05,
"loss": 0.0944,
"step": 570
},
{
"epoch": 1.0450653447498874,
"grad_norm": 4.721229553222656,
"learning_rate": 3.953068592057762e-05,
"loss": 0.1372,
"step": 580
},
{
"epoch": 1.0630914826498423,
"grad_norm": 6.2178955078125,
"learning_rate": 3.935018050541516e-05,
"loss": 0.1412,
"step": 590
},
{
"epoch": 1.0811176205497972,
"grad_norm": 6.32495641708374,
"learning_rate": 3.916967509025271e-05,
"loss": 0.0892,
"step": 600
},
{
"epoch": 1.099143758449752,
"grad_norm": 12.221674919128418,
"learning_rate": 3.898916967509025e-05,
"loss": 0.1333,
"step": 610
},
{
"epoch": 1.117169896349707,
"grad_norm": 1.7088432312011719,
"learning_rate": 3.88086642599278e-05,
"loss": 0.1077,
"step": 620
},
{
"epoch": 1.135196034249662,
"grad_norm": 17.136337280273438,
"learning_rate": 3.862815884476535e-05,
"loss": 0.1381,
"step": 630
},
{
"epoch": 1.153222172149617,
"grad_norm": 5.424814224243164,
"learning_rate": 3.844765342960289e-05,
"loss": 0.1228,
"step": 640
},
{
"epoch": 1.1712483100495719,
"grad_norm": 18.775466918945312,
"learning_rate": 3.826714801444044e-05,
"loss": 0.099,
"step": 650
},
{
"epoch": 1.1892744479495267,
"grad_norm": 8.731025695800781,
"learning_rate": 3.8086642599277976e-05,
"loss": 0.1324,
"step": 660
},
{
"epoch": 1.2073005858494819,
"grad_norm": 9.53122615814209,
"learning_rate": 3.790613718411553e-05,
"loss": 0.1085,
"step": 670
},
{
"epoch": 1.2253267237494367,
"grad_norm": 17.20035743713379,
"learning_rate": 3.7725631768953066e-05,
"loss": 0.0774,
"step": 680
},
{
"epoch": 1.2433528616493916,
"grad_norm": 18.292253494262695,
"learning_rate": 3.754512635379062e-05,
"loss": 0.0965,
"step": 690
},
{
"epoch": 1.2487607030193781,
"eval_f1": 0.7416879795396419,
"eval_loss": 0.10558413714170456,
"eval_precision": 0.6921241050119332,
"eval_recall": 0.7988980716253443,
"eval_runtime": 115.001,
"eval_samples_per_second": 40.634,
"eval_steps_per_second": 5.087,
"step": 693
},
{
"epoch": 1.2613789995493465,
"grad_norm": 6.922005653381348,
"learning_rate": 3.7364620938628155e-05,
"loss": 0.0906,
"step": 700
},
{
"epoch": 1.2794051374493014,
"grad_norm": 33.214107513427734,
"learning_rate": 3.718411552346571e-05,
"loss": 0.072,
"step": 710
},
{
"epoch": 1.2974312753492563,
"grad_norm": 9.25637149810791,
"learning_rate": 3.700361010830325e-05,
"loss": 0.1597,
"step": 720
},
{
"epoch": 1.3154574132492114,
"grad_norm": 10.336318016052246,
"learning_rate": 3.68231046931408e-05,
"loss": 0.1063,
"step": 730
},
{
"epoch": 1.3334835511491663,
"grad_norm": 11.632000923156738,
"learning_rate": 3.664259927797834e-05,
"loss": 0.1175,
"step": 740
},
{
"epoch": 1.3515096890491212,
"grad_norm": 12.909192085266113,
"learning_rate": 3.646209386281589e-05,
"loss": 0.1285,
"step": 750
},
{
"epoch": 1.3695358269490763,
"grad_norm": 7.545075416564941,
"learning_rate": 3.628158844765343e-05,
"loss": 0.0864,
"step": 760
},
{
"epoch": 1.3875619648490312,
"grad_norm": 25.248464584350586,
"learning_rate": 3.610108303249098e-05,
"loss": 0.0769,
"step": 770
},
{
"epoch": 1.405588102748986,
"grad_norm": 10.5347318649292,
"learning_rate": 3.592057761732852e-05,
"loss": 0.088,
"step": 780
},
{
"epoch": 1.423614240648941,
"grad_norm": 13.101734161376953,
"learning_rate": 3.574007220216607e-05,
"loss": 0.1084,
"step": 790
},
{
"epoch": 1.4416403785488958,
"grad_norm": 12.188843727111816,
"learning_rate": 3.555956678700361e-05,
"loss": 0.1101,
"step": 800
},
{
"epoch": 1.4596665164488507,
"grad_norm": 11.552268981933594,
"learning_rate": 3.537906137184116e-05,
"loss": 0.0785,
"step": 810
},
{
"epoch": 1.4776926543488058,
"grad_norm": 10.269721031188965,
"learning_rate": 3.51985559566787e-05,
"loss": 0.1161,
"step": 820
},
{
"epoch": 1.4957187922487607,
"grad_norm": 12.09333610534668,
"learning_rate": 3.5018050541516247e-05,
"loss": 0.1113,
"step": 830
},
{
"epoch": 1.5137449301487156,
"grad_norm": 5.4690141677856445,
"learning_rate": 3.483754512635379e-05,
"loss": 0.1046,
"step": 840
},
{
"epoch": 1.5317710680486707,
"grad_norm": 6.296072959899902,
"learning_rate": 3.4657039711191336e-05,
"loss": 0.1202,
"step": 850
},
{
"epoch": 1.5497972059486256,
"grad_norm": 8.854303359985352,
"learning_rate": 3.447653429602888e-05,
"loss": 0.0929,
"step": 860
},
{
"epoch": 1.5678233438485805,
"grad_norm": 23.987024307250977,
"learning_rate": 3.4296028880866426e-05,
"loss": 0.1326,
"step": 870
},
{
"epoch": 1.5858494817485354,
"grad_norm": 7.229393005371094,
"learning_rate": 3.411552346570397e-05,
"loss": 0.1169,
"step": 880
},
{
"epoch": 1.6038756196484902,
"grad_norm": 5.463610649108887,
"learning_rate": 3.3935018050541516e-05,
"loss": 0.0809,
"step": 890
},
{
"epoch": 1.6219017575484451,
"grad_norm": 5.9659624099731445,
"learning_rate": 3.375451263537907e-05,
"loss": 0.1265,
"step": 900
},
{
"epoch": 1.6399278954484002,
"grad_norm": 17.582469940185547,
"learning_rate": 3.3574007220216606e-05,
"loss": 0.0846,
"step": 910
},
{
"epoch": 1.6579540333483551,
"grad_norm": 13.633155822753906,
"learning_rate": 3.339350180505416e-05,
"loss": 0.0999,
"step": 920
},
{
"epoch": 1.67598017124831,
"grad_norm": 14.496593475341797,
"learning_rate": 3.3212996389891696e-05,
"loss": 0.0903,
"step": 930
},
{
"epoch": 1.694006309148265,
"grad_norm": 7.594356536865234,
"learning_rate": 3.303249097472924e-05,
"loss": 0.1021,
"step": 940
},
{
"epoch": 1.71203244704822,
"grad_norm": 20.628849029541016,
"learning_rate": 3.2851985559566786e-05,
"loss": 0.1131,
"step": 950
},
{
"epoch": 1.7300585849481749,
"grad_norm": 8.25106430053711,
"learning_rate": 3.267148014440433e-05,
"loss": 0.0872,
"step": 960
},
{
"epoch": 1.7480847228481298,
"grad_norm": 17.08070182800293,
"learning_rate": 3.249097472924188e-05,
"loss": 0.0663,
"step": 970
},
{
"epoch": 1.7661108607480847,
"grad_norm": 38.33162307739258,
"learning_rate": 3.231046931407942e-05,
"loss": 0.142,
"step": 980
},
{
"epoch": 1.7841369986480395,
"grad_norm": 17.89957618713379,
"learning_rate": 3.212996389891697e-05,
"loss": 0.0942,
"step": 990
},
{
"epoch": 1.8021631365479944,
"grad_norm": 9.691226959228516,
"learning_rate": 3.194945848375451e-05,
"loss": 0.0902,
"step": 1000
},
{
"epoch": 1.8201892744479495,
"grad_norm": 9.764283180236816,
"learning_rate": 3.176895306859206e-05,
"loss": 0.1108,
"step": 1010
},
{
"epoch": 1.8382154123479044,
"grad_norm": 13.829035758972168,
"learning_rate": 3.15884476534296e-05,
"loss": 0.0881,
"step": 1020
},
{
"epoch": 1.8562415502478595,
"grad_norm": 8.372889518737793,
"learning_rate": 3.140794223826715e-05,
"loss": 0.1272,
"step": 1030
},
{
"epoch": 1.8742676881478144,
"grad_norm": 12.576041221618652,
"learning_rate": 3.12274368231047e-05,
"loss": 0.1076,
"step": 1040
},
{
"epoch": 1.8922938260477693,
"grad_norm": 12.995355606079102,
"learning_rate": 3.104693140794224e-05,
"loss": 0.0932,
"step": 1050
},
{
"epoch": 1.9103199639477242,
"grad_norm": 6.626877784729004,
"learning_rate": 3.086642599277979e-05,
"loss": 0.105,
"step": 1060
},
{
"epoch": 1.928346101847679,
"grad_norm": 11.209733009338379,
"learning_rate": 3.0685920577617325e-05,
"loss": 0.1331,
"step": 1070
},
{
"epoch": 1.946372239747634,
"grad_norm": 7.832246780395508,
"learning_rate": 3.0505415162454877e-05,
"loss": 0.087,
"step": 1080
},
{
"epoch": 1.9643983776475888,
"grad_norm": 9.489190101623535,
"learning_rate": 3.032490974729242e-05,
"loss": 0.0904,
"step": 1090
},
{
"epoch": 1.982424515547544,
"grad_norm": 14.940125465393066,
"learning_rate": 3.0144404332129967e-05,
"loss": 0.0959,
"step": 1100
},
{
"epoch": 2.0,
"grad_norm": 8.826193809509277,
"learning_rate": 2.996389891696751e-05,
"loss": 0.1276,
"step": 1110
}
],
"logging_steps": 10,
"max_steps": 2770,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 3.858514256176742e+16,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}