problem324_model_aug_30 / trainer_state.json
aadityap's picture
Model save
e979846 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 500,
"global_step": 120,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.016666666666666666,
"grad_norm": 0.025269393857377618,
"learning_rate": 4.166666666666667e-06,
"loss": 0.0082,
"step": 1
},
{
"epoch": 0.03333333333333333,
"grad_norm": 0.04867740888947266,
"learning_rate": 8.333333333333334e-06,
"loss": 0.0384,
"step": 2
},
{
"epoch": 0.05,
"grad_norm": 0.03242291180621974,
"learning_rate": 1.25e-05,
"loss": 0.0348,
"step": 3
},
{
"epoch": 0.06666666666666667,
"grad_norm": 0.03136492681683796,
"learning_rate": 1.6666666666666667e-05,
"loss": 0.0196,
"step": 4
},
{
"epoch": 0.08333333333333333,
"grad_norm": 0.02185426746327819,
"learning_rate": 2.0833333333333336e-05,
"loss": 0.0228,
"step": 5
},
{
"epoch": 0.1,
"grad_norm": 0.055190606257933116,
"learning_rate": 2.5e-05,
"loss": 0.0947,
"step": 6
},
{
"epoch": 0.11666666666666667,
"grad_norm": 0.045030318541993665,
"learning_rate": 2.916666666666667e-05,
"loss": 0.0393,
"step": 7
},
{
"epoch": 0.13333333333333333,
"grad_norm": 0.025729698520220775,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0498,
"step": 8
},
{
"epoch": 0.15,
"grad_norm": 0.05452780095219606,
"learning_rate": 3.7500000000000003e-05,
"loss": 0.0388,
"step": 9
},
{
"epoch": 0.16666666666666666,
"grad_norm": 0.040209574901977456,
"learning_rate": 4.166666666666667e-05,
"loss": 0.0311,
"step": 10
},
{
"epoch": 0.18333333333333332,
"grad_norm": 0.036330580825129626,
"learning_rate": 4.5833333333333334e-05,
"loss": 0.0268,
"step": 11
},
{
"epoch": 0.2,
"grad_norm": 0.012117891812127623,
"learning_rate": 5e-05,
"loss": 0.0034,
"step": 12
},
{
"epoch": 0.21666666666666667,
"grad_norm": 0.05827055131143882,
"learning_rate": 4.998942375205502e-05,
"loss": 0.0591,
"step": 13
},
{
"epoch": 0.23333333333333334,
"grad_norm": 0.009703916469083354,
"learning_rate": 4.995770395678171e-05,
"loss": 0.002,
"step": 14
},
{
"epoch": 0.25,
"grad_norm": 0.06529939103039224,
"learning_rate": 4.990486745229364e-05,
"loss": 0.0482,
"step": 15
},
{
"epoch": 0.26666666666666666,
"grad_norm": 0.032574353816168376,
"learning_rate": 4.983095894354858e-05,
"loss": 0.009,
"step": 16
},
{
"epoch": 0.2833333333333333,
"grad_norm": 0.06013001698263978,
"learning_rate": 4.973604096452361e-05,
"loss": 0.0794,
"step": 17
},
{
"epoch": 0.3,
"grad_norm": 0.06898085733425208,
"learning_rate": 4.962019382530521e-05,
"loss": 0.0814,
"step": 18
},
{
"epoch": 0.31666666666666665,
"grad_norm": 0.04013252441989621,
"learning_rate": 4.948351554413879e-05,
"loss": 0.0221,
"step": 19
},
{
"epoch": 0.3333333333333333,
"grad_norm": 0.03654641417934175,
"learning_rate": 4.9326121764495596e-05,
"loss": 0.0265,
"step": 20
},
{
"epoch": 0.35,
"grad_norm": 0.027732402785201702,
"learning_rate": 4.914814565722671e-05,
"loss": 0.0344,
"step": 21
},
{
"epoch": 0.36666666666666664,
"grad_norm": 0.027579526759390415,
"learning_rate": 4.894973780788722e-05,
"loss": 0.0441,
"step": 22
},
{
"epoch": 0.38333333333333336,
"grad_norm": 0.07165522344579221,
"learning_rate": 4.873106608932585e-05,
"loss": 0.1104,
"step": 23
},
{
"epoch": 0.4,
"grad_norm": 0.07867875855803975,
"learning_rate": 4.849231551964771e-05,
"loss": 0.1387,
"step": 24
},
{
"epoch": 0.4166666666666667,
"grad_norm": 0.07261274057887408,
"learning_rate": 4.823368810567056e-05,
"loss": 0.0874,
"step": 25
},
{
"epoch": 0.43333333333333335,
"grad_norm": 0.038717091348658285,
"learning_rate": 4.7955402672006854e-05,
"loss": 0.0076,
"step": 26
},
{
"epoch": 0.45,
"grad_norm": 0.07396842429922766,
"learning_rate": 4.765769467591625e-05,
"loss": 0.0473,
"step": 27
},
{
"epoch": 0.4666666666666667,
"grad_norm": 0.021320518249373258,
"learning_rate": 4.734081600808531e-05,
"loss": 0.0285,
"step": 28
},
{
"epoch": 0.48333333333333334,
"grad_norm": 0.04526100183831436,
"learning_rate": 4.700503477950278e-05,
"loss": 0.0273,
"step": 29
},
{
"epoch": 0.5,
"grad_norm": 0.04899249641256801,
"learning_rate": 4.665063509461097e-05,
"loss": 0.0471,
"step": 30
},
{
"epoch": 0.5166666666666667,
"grad_norm": 0.052899053939905495,
"learning_rate": 4.627791681092499e-05,
"loss": 0.0656,
"step": 31
},
{
"epoch": 0.5333333333333333,
"grad_norm": 0.006516498320631479,
"learning_rate": 4.588719528532342e-05,
"loss": 0.0015,
"step": 32
},
{
"epoch": 0.55,
"grad_norm": 0.027585766138034963,
"learning_rate": 4.54788011072248e-05,
"loss": 0.017,
"step": 33
},
{
"epoch": 0.5666666666666667,
"grad_norm": 0.03367120895796786,
"learning_rate": 4.50530798188761e-05,
"loss": 0.0131,
"step": 34
},
{
"epoch": 0.5833333333333334,
"grad_norm": 0.05439044379589846,
"learning_rate": 4.4610391622989396e-05,
"loss": 0.0286,
"step": 35
},
{
"epoch": 0.6,
"grad_norm": 0.024693975957964298,
"learning_rate": 4.415111107797445e-05,
"loss": 0.0141,
"step": 36
},
{
"epoch": 0.6166666666666667,
"grad_norm": 0.06746654433419255,
"learning_rate": 4.36756267810249e-05,
"loss": 0.0169,
"step": 37
},
{
"epoch": 0.6333333333333333,
"grad_norm": 0.014165653821600677,
"learning_rate": 4.318434103932622e-05,
"loss": 0.0052,
"step": 38
},
{
"epoch": 0.65,
"grad_norm": 0.07632703997890092,
"learning_rate": 4.267766952966369e-05,
"loss": 0.0471,
"step": 39
},
{
"epoch": 0.6666666666666666,
"grad_norm": 0.0727379407724865,
"learning_rate": 4.215604094671835e-05,
"loss": 0.0218,
"step": 40
},
{
"epoch": 0.6833333333333333,
"grad_norm": 0.017931284971321647,
"learning_rate": 4.1619896640348445e-05,
"loss": 0.0132,
"step": 41
},
{
"epoch": 0.7,
"grad_norm": 0.032489475949465524,
"learning_rate": 4.1069690242163484e-05,
"loss": 0.0037,
"step": 42
},
{
"epoch": 0.7166666666666667,
"grad_norm": 0.036765464462579944,
"learning_rate": 4.05058872817065e-05,
"loss": 0.0172,
"step": 43
},
{
"epoch": 0.7333333333333333,
"grad_norm": 0.04363017416856556,
"learning_rate": 3.9928964792569655e-05,
"loss": 0.0313,
"step": 44
},
{
"epoch": 0.75,
"grad_norm": 0.032482425980983,
"learning_rate": 3.933941090877615e-05,
"loss": 0.0132,
"step": 45
},
{
"epoch": 0.7666666666666667,
"grad_norm": 0.024699611173539596,
"learning_rate": 3.873772445177015e-05,
"loss": 0.0152,
"step": 46
},
{
"epoch": 0.7833333333333333,
"grad_norm": 0.06857369891564866,
"learning_rate": 3.8124414508364e-05,
"loss": 0.0397,
"step": 47
},
{
"epoch": 0.8,
"grad_norm": 0.02093165466812801,
"learning_rate": 3.7500000000000003e-05,
"loss": 0.0136,
"step": 48
},
{
"epoch": 0.8166666666666667,
"grad_norm": 0.05002920705190205,
"learning_rate": 3.686500924369101e-05,
"loss": 0.0274,
"step": 49
},
{
"epoch": 0.8333333333333334,
"grad_norm": 0.06589372830420803,
"learning_rate": 3.621997950501156e-05,
"loss": 0.031,
"step": 50
},
{
"epoch": 0.85,
"grad_norm": 0.022743333709719687,
"learning_rate": 3.556545654351749e-05,
"loss": 0.0116,
"step": 51
},
{
"epoch": 0.8666666666666667,
"grad_norm": 0.04233508499490824,
"learning_rate": 3.490199415097892e-05,
"loss": 0.0251,
"step": 52
},
{
"epoch": 0.8833333333333333,
"grad_norm": 0.05800525727327983,
"learning_rate": 3.423015368281711e-05,
"loss": 0.0088,
"step": 53
},
{
"epoch": 0.9,
"grad_norm": 0.03787022881730521,
"learning_rate": 3.355050358314172e-05,
"loss": 0.0199,
"step": 54
},
{
"epoch": 0.9166666666666666,
"grad_norm": 0.029562295638423525,
"learning_rate": 3.2863618903790346e-05,
"loss": 0.0079,
"step": 55
},
{
"epoch": 0.9333333333333333,
"grad_norm": 0.008415241622389717,
"learning_rate": 3.217008081777726e-05,
"loss": 0.0016,
"step": 56
},
{
"epoch": 0.95,
"grad_norm": 0.00982931157684509,
"learning_rate": 3.147047612756302e-05,
"loss": 0.0022,
"step": 57
},
{
"epoch": 0.9666666666666667,
"grad_norm": 0.02791032608951001,
"learning_rate": 3.076539676856101e-05,
"loss": 0.0065,
"step": 58
},
{
"epoch": 0.9833333333333333,
"grad_norm": 0.010409742361526964,
"learning_rate": 3.0055439308300952e-05,
"loss": 0.0019,
"step": 59
},
{
"epoch": 1.0,
"grad_norm": 0.05998312530353288,
"learning_rate": 2.9341204441673266e-05,
"loss": 0.0186,
"step": 60
},
{
"epoch": 1.0,
"eval_loss": 0.023462090641260147,
"eval_runtime": 0.4163,
"eval_samples_per_second": 2.402,
"eval_steps_per_second": 2.402,
"step": 60
},
{
"epoch": 1.0166666666666666,
"grad_norm": 0.009620282142640397,
"learning_rate": 2.8623296482681166e-05,
"loss": 0.0017,
"step": 61
},
{
"epoch": 1.0333333333333334,
"grad_norm": 0.010414129384870628,
"learning_rate": 2.7902322853130757e-05,
"loss": 0.002,
"step": 62
},
{
"epoch": 1.05,
"grad_norm": 0.028335054385269624,
"learning_rate": 2.717889356869146e-05,
"loss": 0.0137,
"step": 63
},
{
"epoch": 1.0666666666666667,
"grad_norm": 0.006810512877725842,
"learning_rate": 2.6453620722761896e-05,
"loss": 0.0013,
"step": 64
},
{
"epoch": 1.0833333333333333,
"grad_norm": 0.052553779140008625,
"learning_rate": 2.5727117968577784e-05,
"loss": 0.0236,
"step": 65
},
{
"epoch": 1.1,
"grad_norm": 0.02628237526317913,
"learning_rate": 2.5e-05,
"loss": 0.003,
"step": 66
},
{
"epoch": 1.1166666666666667,
"grad_norm": 0.024678762077929545,
"learning_rate": 2.4272882031422215e-05,
"loss": 0.0089,
"step": 67
},
{
"epoch": 1.1333333333333333,
"grad_norm": 0.005679579845213851,
"learning_rate": 2.3546379277238107e-05,
"loss": 0.0004,
"step": 68
},
{
"epoch": 1.15,
"grad_norm": 0.01877750041666978,
"learning_rate": 2.2821106431308544e-05,
"loss": 0.0084,
"step": 69
},
{
"epoch": 1.1666666666666667,
"grad_norm": 0.013255608958773757,
"learning_rate": 2.2097677146869242e-05,
"loss": 0.0036,
"step": 70
},
{
"epoch": 1.1833333333333333,
"grad_norm": 0.02437149288050262,
"learning_rate": 2.1376703517318837e-05,
"loss": 0.0036,
"step": 71
},
{
"epoch": 1.2,
"grad_norm": 0.05973693020145483,
"learning_rate": 2.0658795558326743e-05,
"loss": 0.016,
"step": 72
},
{
"epoch": 1.2166666666666668,
"grad_norm": 0.026031457093812112,
"learning_rate": 1.9944560691699057e-05,
"loss": 0.0059,
"step": 73
},
{
"epoch": 1.2333333333333334,
"grad_norm": 0.02173019223927693,
"learning_rate": 1.9234603231438995e-05,
"loss": 0.0088,
"step": 74
},
{
"epoch": 1.25,
"grad_norm": 0.02269186282491817,
"learning_rate": 1.852952387243698e-05,
"loss": 0.0118,
"step": 75
},
{
"epoch": 1.2666666666666666,
"grad_norm": 0.06675557782721865,
"learning_rate": 1.7829919182222752e-05,
"loss": 0.0309,
"step": 76
},
{
"epoch": 1.2833333333333332,
"grad_norm": 0.005662660244441698,
"learning_rate": 1.7136381096209664e-05,
"loss": 0.0009,
"step": 77
},
{
"epoch": 1.3,
"grad_norm": 0.05008471620600304,
"learning_rate": 1.6449496416858284e-05,
"loss": 0.0207,
"step": 78
},
{
"epoch": 1.3166666666666667,
"grad_norm": 0.004266332439973932,
"learning_rate": 1.5769846317182893e-05,
"loss": 0.0007,
"step": 79
},
{
"epoch": 1.3333333333333333,
"grad_norm": 0.03129098476091339,
"learning_rate": 1.509800584902108e-05,
"loss": 0.0051,
"step": 80
},
{
"epoch": 1.35,
"grad_norm": 0.0061920540760470365,
"learning_rate": 1.443454345648252e-05,
"loss": 0.0007,
"step": 81
},
{
"epoch": 1.3666666666666667,
"grad_norm": 0.027460901731622724,
"learning_rate": 1.3780020494988446e-05,
"loss": 0.0097,
"step": 82
},
{
"epoch": 1.3833333333333333,
"grad_norm": 0.022898768868336757,
"learning_rate": 1.313499075630899e-05,
"loss": 0.0016,
"step": 83
},
{
"epoch": 1.4,
"grad_norm": 0.0021036670493289997,
"learning_rate": 1.2500000000000006e-05,
"loss": 0.0004,
"step": 84
},
{
"epoch": 1.4166666666666667,
"grad_norm": 0.008070628993667719,
"learning_rate": 1.1875585491636e-05,
"loss": 0.0014,
"step": 85
},
{
"epoch": 1.4333333333333333,
"grad_norm": 0.013311467116804818,
"learning_rate": 1.126227554822985e-05,
"loss": 0.003,
"step": 86
},
{
"epoch": 1.45,
"grad_norm": 0.007737229021259105,
"learning_rate": 1.0660589091223855e-05,
"loss": 0.0008,
"step": 87
},
{
"epoch": 1.4666666666666668,
"grad_norm": 0.060519449831449114,
"learning_rate": 1.0071035207430352e-05,
"loss": 0.0245,
"step": 88
},
{
"epoch": 1.4833333333333334,
"grad_norm": 0.03956824192763825,
"learning_rate": 9.494112718293501e-06,
"loss": 0.0037,
"step": 89
},
{
"epoch": 1.5,
"grad_norm": 0.029455349206802237,
"learning_rate": 8.930309757836517e-06,
"loss": 0.0176,
"step": 90
},
{
"epoch": 1.5166666666666666,
"grad_norm": 0.09204417580838978,
"learning_rate": 8.380103359651553e-06,
"loss": 0.053,
"step": 91
},
{
"epoch": 1.5333333333333332,
"grad_norm": 0.029342253202570764,
"learning_rate": 7.843959053281663e-06,
"loss": 0.0114,
"step": 92
},
{
"epoch": 1.55,
"grad_norm": 0.04653111352863076,
"learning_rate": 7.3223304703363135e-06,
"loss": 0.0183,
"step": 93
},
{
"epoch": 1.5666666666666667,
"grad_norm": 0.045617946414850744,
"learning_rate": 6.815658960673782e-06,
"loss": 0.0034,
"step": 94
},
{
"epoch": 1.5833333333333335,
"grad_norm": 0.06025932248044368,
"learning_rate": 6.324373218975105e-06,
"loss": 0.0032,
"step": 95
},
{
"epoch": 1.6,
"grad_norm": 0.02712157694660743,
"learning_rate": 5.848888922025553e-06,
"loss": 0.0074,
"step": 96
},
{
"epoch": 1.6166666666666667,
"grad_norm": 0.028507426580009265,
"learning_rate": 5.389608377010608e-06,
"loss": 0.0104,
"step": 97
},
{
"epoch": 1.6333333333333333,
"grad_norm": 0.03533536426677586,
"learning_rate": 4.946920181123904e-06,
"loss": 0.0057,
"step": 98
},
{
"epoch": 1.65,
"grad_norm": 0.01427909957453889,
"learning_rate": 4.521198892775203e-06,
"loss": 0.0019,
"step": 99
},
{
"epoch": 1.6666666666666665,
"grad_norm": 0.06426857718427728,
"learning_rate": 4.112804714676594e-06,
"loss": 0.028,
"step": 100
},
{
"epoch": 1.6833333333333333,
"grad_norm": 0.025113714087324157,
"learning_rate": 3.7220831890750067e-06,
"loss": 0.0079,
"step": 101
},
{
"epoch": 1.7,
"grad_norm": 0.06272910859633025,
"learning_rate": 3.3493649053890326e-06,
"loss": 0.0289,
"step": 102
},
{
"epoch": 1.7166666666666668,
"grad_norm": 0.0664046761556785,
"learning_rate": 2.9949652204972254e-06,
"loss": 0.0073,
"step": 103
},
{
"epoch": 1.7333333333333334,
"grad_norm": 0.003990758285364662,
"learning_rate": 2.659183991914696e-06,
"loss": 0.0006,
"step": 104
},
{
"epoch": 1.75,
"grad_norm": 0.03506550328085753,
"learning_rate": 2.3423053240837515e-06,
"loss": 0.0146,
"step": 105
},
{
"epoch": 1.7666666666666666,
"grad_norm": 0.03652102411445378,
"learning_rate": 2.044597327993153e-06,
"loss": 0.0131,
"step": 106
},
{
"epoch": 1.7833333333333332,
"grad_norm": 0.02468495241018833,
"learning_rate": 1.7663118943294366e-06,
"loss": 0.0049,
"step": 107
},
{
"epoch": 1.8,
"grad_norm": 0.03187314902879643,
"learning_rate": 1.5076844803522922e-06,
"loss": 0.0107,
"step": 108
},
{
"epoch": 1.8166666666666667,
"grad_norm": 0.012137389269953076,
"learning_rate": 1.2689339106741527e-06,
"loss": 0.0023,
"step": 109
},
{
"epoch": 1.8333333333333335,
"grad_norm": 0.02667426848176036,
"learning_rate": 1.0502621921127776e-06,
"loss": 0.0126,
"step": 110
},
{
"epoch": 1.85,
"grad_norm": 0.024304629326650305,
"learning_rate": 8.51854342773295e-07,
"loss": 0.0014,
"step": 111
},
{
"epoch": 1.8666666666666667,
"grad_norm": 0.027987508604913596,
"learning_rate": 6.738782355044049e-07,
"loss": 0.0085,
"step": 112
},
{
"epoch": 1.8833333333333333,
"grad_norm": 0.007778146112293743,
"learning_rate": 5.164844558612131e-07,
"loss": 0.0009,
"step": 113
},
{
"epoch": 1.9,
"grad_norm": 0.01542814885568116,
"learning_rate": 3.7980617469479953e-07,
"loss": 0.0039,
"step": 114
},
{
"epoch": 1.9166666666666665,
"grad_norm": 0.0022728161341522934,
"learning_rate": 2.6395903547638825e-07,
"loss": 0.0003,
"step": 115
},
{
"epoch": 1.9333333333333333,
"grad_norm": 0.041240776754602505,
"learning_rate": 1.6904105645142444e-07,
"loss": 0.0148,
"step": 116
},
{
"epoch": 1.95,
"grad_norm": 0.013737757216489009,
"learning_rate": 9.513254770636137e-08,
"loss": 0.0039,
"step": 117
},
{
"epoch": 1.9666666666666668,
"grad_norm": 0.003402223050457768,
"learning_rate": 4.229604321829561e-08,
"loss": 0.0004,
"step": 118
},
{
"epoch": 1.9833333333333334,
"grad_norm": 0.016913072535640866,
"learning_rate": 1.0576247944985018e-08,
"loss": 0.0028,
"step": 119
},
{
"epoch": 2.0,
"grad_norm": 0.0530401508275478,
"learning_rate": 0.0,
"loss": 0.0147,
"step": 120
},
{
"epoch": 2.0,
"eval_loss": 0.026322441175580025,
"eval_runtime": 1.4216,
"eval_samples_per_second": 0.703,
"eval_steps_per_second": 0.703,
"step": 120
},
{
"epoch": 2.0,
"step": 120,
"total_flos": 1446028247040.0,
"train_loss": 0.019833612175716553,
"train_runtime": 384.9879,
"train_samples_per_second": 0.623,
"train_steps_per_second": 0.312
}
],
"logging_steps": 1,
"max_steps": 120,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 1446028247040.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}