aa57fe3709
Model: PKU-Alignment/ProgressGym-HistLlama3-8B-C013-pretrain-v0.2 Source: Original Platform
841 lines
21 KiB
JSON
841 lines
21 KiB
JSON
{
|
|
"best_metric": 1.5942710638046265,
|
|
"best_model_checkpoint": "./output/training_results/C013_llama3-8b-base_pretrain_20240428_005832/checkpoint-65",
|
|
"epoch": 4.0,
|
|
"eval_steps": 5,
|
|
"global_step": 268,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.014925373134328358,
|
|
"grad_norm": 0.0,
|
|
"learning_rate": 0.0,
|
|
"loss": 1.7594,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.014925373134328358,
|
|
"eval_loss": 1.7162834405899048,
|
|
"eval_runtime": 5.9233,
|
|
"eval_samples_per_second": 80.36,
|
|
"eval_steps_per_second": 0.675,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.07462686567164178,
|
|
"grad_norm": 3.233517003168196,
|
|
"learning_rate": 2.25e-06,
|
|
"loss": 1.7333,
|
|
"step": 5
|
|
},
|
|
{
|
|
"epoch": 0.07462686567164178,
|
|
"eval_loss": 1.7008264064788818,
|
|
"eval_runtime": 5.9309,
|
|
"eval_samples_per_second": 80.257,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 5
|
|
},
|
|
{
|
|
"epoch": 0.14925373134328357,
|
|
"grad_norm": 3.423591914627149,
|
|
"learning_rate": 6e-06,
|
|
"loss": 1.6854,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.14925373134328357,
|
|
"eval_loss": 1.682459831237793,
|
|
"eval_runtime": 5.9184,
|
|
"eval_samples_per_second": 80.427,
|
|
"eval_steps_per_second": 0.676,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.22388059701492538,
|
|
"grad_norm": 2.75139227215463,
|
|
"learning_rate": 9.75e-06,
|
|
"loss": 1.6897,
|
|
"step": 15
|
|
},
|
|
{
|
|
"epoch": 0.22388059701492538,
|
|
"eval_loss": 1.670121192932129,
|
|
"eval_runtime": 5.9473,
|
|
"eval_samples_per_second": 80.036,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 15
|
|
},
|
|
{
|
|
"epoch": 0.29850746268656714,
|
|
"grad_norm": 2.55059243594346,
|
|
"learning_rate": 1.3500000000000001e-05,
|
|
"loss": 1.6656,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.29850746268656714,
|
|
"eval_loss": 1.6651127338409424,
|
|
"eval_runtime": 5.942,
|
|
"eval_samples_per_second": 80.108,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.373134328358209,
|
|
"grad_norm": 2.7488276843880652,
|
|
"learning_rate": 1.3725993373990923e-05,
|
|
"loss": 1.7254,
|
|
"step": 25
|
|
},
|
|
{
|
|
"epoch": 0.373134328358209,
|
|
"eval_loss": 1.6679344177246094,
|
|
"eval_runtime": 5.9726,
|
|
"eval_samples_per_second": 79.697,
|
|
"eval_steps_per_second": 0.67,
|
|
"step": 25
|
|
},
|
|
{
|
|
"epoch": 0.44776119402985076,
|
|
"grad_norm": 3.497477848836299,
|
|
"learning_rate": 1.096088986215196e-05,
|
|
"loss": 1.7178,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.44776119402985076,
|
|
"eval_loss": 1.6541944742202759,
|
|
"eval_runtime": 5.953,
|
|
"eval_samples_per_second": 79.96,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.5223880597014925,
|
|
"grad_norm": 2.7958991196951666,
|
|
"learning_rate": 8.713726547724396e-06,
|
|
"loss": 1.6656,
|
|
"step": 35
|
|
},
|
|
{
|
|
"epoch": 0.5223880597014925,
|
|
"eval_loss": 1.6459277868270874,
|
|
"eval_runtime": 5.9488,
|
|
"eval_samples_per_second": 80.016,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 35
|
|
},
|
|
{
|
|
"epoch": 0.5970149253731343,
|
|
"grad_norm": 2.2395806127723907,
|
|
"learning_rate": 6.8954890860223175e-06,
|
|
"loss": 1.6647,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.5970149253731343,
|
|
"eval_loss": 1.6308344602584839,
|
|
"eval_runtime": 5.9441,
|
|
"eval_samples_per_second": 80.08,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.6716417910447762,
|
|
"grad_norm": 2.542267170825853,
|
|
"learning_rate": 5.431031497828169e-06,
|
|
"loss": 1.6645,
|
|
"step": 45
|
|
},
|
|
{
|
|
"epoch": 0.6716417910447762,
|
|
"eval_loss": 1.6204748153686523,
|
|
"eval_runtime": 5.9555,
|
|
"eval_samples_per_second": 79.927,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 45
|
|
},
|
|
{
|
|
"epoch": 0.746268656716418,
|
|
"grad_norm": 2.796577480411167,
|
|
"learning_rate": 4.257147264863097e-06,
|
|
"loss": 1.6151,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.746268656716418,
|
|
"eval_loss": 1.6129424571990967,
|
|
"eval_runtime": 5.9792,
|
|
"eval_samples_per_second": 79.609,
|
|
"eval_steps_per_second": 0.669,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.8208955223880597,
|
|
"grad_norm": 2.8952316281119757,
|
|
"learning_rate": 3.320875141743125e-06,
|
|
"loss": 1.6359,
|
|
"step": 55
|
|
},
|
|
{
|
|
"epoch": 0.8208955223880597,
|
|
"eval_loss": 1.6052401065826416,
|
|
"eval_runtime": 5.9501,
|
|
"eval_samples_per_second": 79.998,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 55
|
|
},
|
|
{
|
|
"epoch": 0.8955223880597015,
|
|
"grad_norm": 2.455450911279608,
|
|
"learning_rate": 2.57801538033309e-06,
|
|
"loss": 1.5885,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 0.8955223880597015,
|
|
"eval_loss": 1.5995320081710815,
|
|
"eval_runtime": 5.9574,
|
|
"eval_samples_per_second": 79.9,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 0.9701492537313433,
|
|
"grad_norm": 1.8236330059926131,
|
|
"learning_rate": 1.9918341191886553e-06,
|
|
"loss": 1.6142,
|
|
"step": 65
|
|
},
|
|
{
|
|
"epoch": 0.9701492537313433,
|
|
"eval_loss": 1.5942710638046265,
|
|
"eval_runtime": 5.9491,
|
|
"eval_samples_per_second": 80.013,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 65
|
|
},
|
|
{
|
|
"epoch": 1.044776119402985,
|
|
"grad_norm": 2.5122250154998893,
|
|
"learning_rate": 1.5319356141270557e-06,
|
|
"loss": 1.4875,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 1.044776119402985,
|
|
"eval_loss": 1.5963352918624878,
|
|
"eval_runtime": 5.9465,
|
|
"eval_samples_per_second": 80.047,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 1.1194029850746268,
|
|
"grad_norm": 2.2374216929111017,
|
|
"learning_rate": 1.173283781790582e-06,
|
|
"loss": 1.3844,
|
|
"step": 75
|
|
},
|
|
{
|
|
"epoch": 1.1194029850746268,
|
|
"eval_loss": 1.611823558807373,
|
|
"eval_runtime": 5.9845,
|
|
"eval_samples_per_second": 79.538,
|
|
"eval_steps_per_second": 0.668,
|
|
"step": 75
|
|
},
|
|
{
|
|
"epoch": 1.1940298507462686,
|
|
"grad_norm": 2.061088772790073,
|
|
"learning_rate": 8.95356202130823e-07,
|
|
"loss": 1.3555,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 1.1940298507462686,
|
|
"eval_loss": 1.6069068908691406,
|
|
"eval_runtime": 5.9508,
|
|
"eval_samples_per_second": 79.99,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 1.2686567164179103,
|
|
"grad_norm": 2.0165800449063878,
|
|
"learning_rate": 6.814152836799623e-07,
|
|
"loss": 1.3597,
|
|
"step": 85
|
|
},
|
|
{
|
|
"epoch": 1.2686567164179103,
|
|
"eval_loss": 1.6040061712265015,
|
|
"eval_runtime": 5.9381,
|
|
"eval_samples_per_second": 80.161,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 85
|
|
},
|
|
{
|
|
"epoch": 1.3432835820895521,
|
|
"grad_norm": 2.0046777857294593,
|
|
"learning_rate": 5.178827427887253e-07,
|
|
"loss": 1.3737,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 1.3432835820895521,
|
|
"eval_loss": 1.607108235359192,
|
|
"eval_runtime": 5.938,
|
|
"eval_samples_per_second": 80.161,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 1.417910447761194,
|
|
"grad_norm": 1.9599395887640554,
|
|
"learning_rate": 3.938048900654646e-07,
|
|
"loss": 1.3492,
|
|
"step": 95
|
|
},
|
|
{
|
|
"epoch": 1.417910447761194,
|
|
"eval_loss": 1.607422113418579,
|
|
"eval_runtime": 5.9586,
|
|
"eval_samples_per_second": 79.884,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 95
|
|
},
|
|
{
|
|
"epoch": 1.4925373134328357,
|
|
"grad_norm": 2.090141489085601,
|
|
"learning_rate": 3.0039745928104856e-07,
|
|
"loss": 1.3826,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 1.4925373134328357,
|
|
"eval_loss": 1.6055359840393066,
|
|
"eval_runtime": 5.9265,
|
|
"eval_samples_per_second": 80.317,
|
|
"eval_steps_per_second": 0.675,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 1.5671641791044775,
|
|
"grad_norm": 1.9332970593875183,
|
|
"learning_rate": 2.3065986110942366e-07,
|
|
"loss": 1.3533,
|
|
"step": 105
|
|
},
|
|
{
|
|
"epoch": 1.5671641791044775,
|
|
"eval_loss": 1.6034619808197021,
|
|
"eval_runtime": 5.9297,
|
|
"eval_samples_per_second": 80.273,
|
|
"eval_steps_per_second": 0.675,
|
|
"step": 105
|
|
},
|
|
{
|
|
"epoch": 1.6417910447761193,
|
|
"grad_norm": 1.942043821463108,
|
|
"learning_rate": 1.790498012203728e-07,
|
|
"loss": 1.3611,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 1.6417910447761193,
|
|
"eval_loss": 1.6023228168487549,
|
|
"eval_runtime": 5.9407,
|
|
"eval_samples_per_second": 80.126,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 1.716417910447761,
|
|
"grad_norm": 1.9221561505361804,
|
|
"learning_rate": 1.412101742620585e-07,
|
|
"loss": 1.328,
|
|
"step": 115
|
|
},
|
|
{
|
|
"epoch": 1.716417910447761,
|
|
"eval_loss": 1.602164387702942,
|
|
"eval_runtime": 5.9392,
|
|
"eval_samples_per_second": 80.145,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 115
|
|
},
|
|
{
|
|
"epoch": 1.7910447761194028,
|
|
"grad_norm": 1.8559815029587312,
|
|
"learning_rate": 1.1374103686567517e-07,
|
|
"loss": 1.3443,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 1.7910447761194028,
|
|
"eval_loss": 1.602574348449707,
|
|
"eval_runtime": 5.9713,
|
|
"eval_samples_per_second": 79.715,
|
|
"eval_steps_per_second": 0.67,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 1.8656716417910446,
|
|
"grad_norm": 2.031347067054233,
|
|
"learning_rate": 9.401027853897119e-08,
|
|
"loss": 1.3386,
|
|
"step": 125
|
|
},
|
|
{
|
|
"epoch": 1.8656716417910446,
|
|
"eval_loss": 1.602864146232605,
|
|
"eval_runtime": 5.956,
|
|
"eval_samples_per_second": 79.919,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 125
|
|
},
|
|
{
|
|
"epoch": 1.9402985074626866,
|
|
"grad_norm": 1.9772043505318835,
|
|
"learning_rate": 7.999735362178058e-08,
|
|
"loss": 1.3396,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 1.9402985074626866,
|
|
"eval_loss": 1.6029260158538818,
|
|
"eval_runtime": 5.955,
|
|
"eval_samples_per_second": 79.933,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 2.014925373134328,
|
|
"grad_norm": 2.0413931319541203,
|
|
"learning_rate": 7.016511465398424e-08,
|
|
"loss": 1.3573,
|
|
"step": 135
|
|
},
|
|
{
|
|
"epoch": 2.014925373134328,
|
|
"eval_loss": 1.6028627157211304,
|
|
"eval_runtime": 5.974,
|
|
"eval_samples_per_second": 79.679,
|
|
"eval_steps_per_second": 0.67,
|
|
"step": 135
|
|
},
|
|
{
|
|
"epoch": 2.08955223880597,
|
|
"grad_norm": 1.9174789145055136,
|
|
"learning_rate": 6.335540171983123e-08,
|
|
"loss": 1.3754,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 2.08955223880597,
|
|
"eval_loss": 1.6034482717514038,
|
|
"eval_runtime": 5.9455,
|
|
"eval_samples_per_second": 80.061,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 2.1641791044776117,
|
|
"grad_norm": 1.8365955877771794,
|
|
"learning_rate": 5.870459761414777e-08,
|
|
"loss": 1.3229,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 2.1641791044776117,
|
|
"eval_loss": 1.6044023036956787,
|
|
"eval_runtime": 5.9418,
|
|
"eval_samples_per_second": 80.11,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 145
|
|
},
|
|
{
|
|
"epoch": 2.2388059701492535,
|
|
"grad_norm": 1.957661723231933,
|
|
"learning_rate": 5.5575858924021805e-08,
|
|
"loss": 1.3194,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 2.2388059701492535,
|
|
"eval_loss": 1.60551917552948,
|
|
"eval_runtime": 5.9349,
|
|
"eval_samples_per_second": 80.204,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 2.3134328358208958,
|
|
"grad_norm": 1.910367302653755,
|
|
"learning_rate": 5.3505182098899865e-08,
|
|
"loss": 1.3361,
|
|
"step": 155
|
|
},
|
|
{
|
|
"epoch": 2.3134328358208958,
|
|
"eval_loss": 1.6064879894256592,
|
|
"eval_runtime": 5.9543,
|
|
"eval_samples_per_second": 79.942,
|
|
"eval_steps_per_second": 0.672,
|
|
"step": 155
|
|
},
|
|
{
|
|
"epoch": 2.388059701492537,
|
|
"grad_norm": 1.978379692085088,
|
|
"learning_rate": 5.2158864924030916e-08,
|
|
"loss": 1.3231,
|
|
"step": 160
|
|
},
|
|
{
|
|
"epoch": 2.388059701492537,
|
|
"eval_loss": 1.60716712474823,
|
|
"eval_runtime": 5.959,
|
|
"eval_samples_per_second": 79.879,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 160
|
|
},
|
|
{
|
|
"epoch": 2.4626865671641793,
|
|
"grad_norm": 1.8179876847012644,
|
|
"learning_rate": 5.130028101466846e-08,
|
|
"loss": 1.32,
|
|
"step": 165
|
|
},
|
|
{
|
|
"epoch": 2.4626865671641793,
|
|
"eval_loss": 1.6076022386550903,
|
|
"eval_runtime": 5.918,
|
|
"eval_samples_per_second": 80.433,
|
|
"eval_steps_per_second": 0.676,
|
|
"step": 165
|
|
},
|
|
{
|
|
"epoch": 2.5373134328358207,
|
|
"grad_norm": 1.947999010708536,
|
|
"learning_rate": 5.076420137573488e-08,
|
|
"loss": 1.3406,
|
|
"step": 170
|
|
},
|
|
{
|
|
"epoch": 2.5373134328358207,
|
|
"eval_loss": 1.6078119277954102,
|
|
"eval_runtime": 5.9402,
|
|
"eval_samples_per_second": 80.132,
|
|
"eval_steps_per_second": 0.673,
|
|
"step": 170
|
|
},
|
|
{
|
|
"epoch": 2.611940298507463,
|
|
"grad_norm": 1.9301332893200993,
|
|
"learning_rate": 5.0437175954586924e-08,
|
|
"loss": 1.3184,
|
|
"step": 175
|
|
},
|
|
{
|
|
"epoch": 2.611940298507463,
|
|
"eval_loss": 1.6078969240188599,
|
|
"eval_runtime": 5.9794,
|
|
"eval_samples_per_second": 79.607,
|
|
"eval_steps_per_second": 0.669,
|
|
"step": 175
|
|
},
|
|
{
|
|
"epoch": 2.6865671641791042,
|
|
"grad_norm": 1.8025714970137283,
|
|
"learning_rate": 5.024273255042209e-08,
|
|
"loss": 1.2745,
|
|
"step": 180
|
|
},
|
|
{
|
|
"epoch": 2.6865671641791042,
|
|
"eval_loss": 1.607950210571289,
|
|
"eval_runtime": 5.9775,
|
|
"eval_samples_per_second": 79.632,
|
|
"eval_steps_per_second": 0.669,
|
|
"step": 180
|
|
},
|
|
{
|
|
"epoch": 2.7611940298507465,
|
|
"grad_norm": 1.8246857333350315,
|
|
"learning_rate": 5.013036339654868e-08,
|
|
"loss": 1.3024,
|
|
"step": 185
|
|
},
|
|
{
|
|
"epoch": 2.7611940298507465,
|
|
"eval_loss": 1.607932448387146,
|
|
"eval_runtime": 5.9386,
|
|
"eval_samples_per_second": 80.154,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 185
|
|
},
|
|
{
|
|
"epoch": 2.835820895522388,
|
|
"grad_norm": 1.8544541968240376,
|
|
"learning_rate": 5.006745403152126e-08,
|
|
"loss": 1.3243,
|
|
"step": 190
|
|
},
|
|
{
|
|
"epoch": 2.835820895522388,
|
|
"eval_loss": 1.6079127788543701,
|
|
"eval_runtime": 5.9362,
|
|
"eval_samples_per_second": 80.186,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 190
|
|
},
|
|
{
|
|
"epoch": 2.91044776119403,
|
|
"grad_norm": 1.9345137440066773,
|
|
"learning_rate": 5.0033467418823265e-08,
|
|
"loss": 1.3239,
|
|
"step": 195
|
|
},
|
|
{
|
|
"epoch": 2.91044776119403,
|
|
"eval_loss": 1.6080163717269897,
|
|
"eval_runtime": 5.9581,
|
|
"eval_samples_per_second": 79.892,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 195
|
|
},
|
|
{
|
|
"epoch": 2.9850746268656714,
|
|
"grad_norm": 2.033807758057437,
|
|
"learning_rate": 5.001583122566221e-08,
|
|
"loss": 1.3349,
|
|
"step": 200
|
|
},
|
|
{
|
|
"epoch": 2.9850746268656714,
|
|
"eval_loss": 1.6080641746520996,
|
|
"eval_runtime": 5.9287,
|
|
"eval_samples_per_second": 80.287,
|
|
"eval_steps_per_second": 0.675,
|
|
"step": 200
|
|
},
|
|
{
|
|
"epoch": 3.0597014925373136,
|
|
"grad_norm": 1.923714106710108,
|
|
"learning_rate": 5.0007090159326376e-08,
|
|
"loss": 1.337,
|
|
"step": 205
|
|
},
|
|
{
|
|
"epoch": 3.0597014925373136,
|
|
"eval_loss": 1.6079014539718628,
|
|
"eval_runtime": 5.9355,
|
|
"eval_samples_per_second": 80.195,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 205
|
|
},
|
|
{
|
|
"epoch": 3.1343283582089554,
|
|
"grad_norm": 1.8059505650037335,
|
|
"learning_rate": 5.000298058072559e-08,
|
|
"loss": 1.3091,
|
|
"step": 210
|
|
},
|
|
{
|
|
"epoch": 3.1343283582089554,
|
|
"eval_loss": 1.6077641248703003,
|
|
"eval_runtime": 5.9641,
|
|
"eval_samples_per_second": 79.811,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 210
|
|
},
|
|
{
|
|
"epoch": 3.208955223880597,
|
|
"grad_norm": 1.894523950936326,
|
|
"learning_rate": 5.0001163431988796e-08,
|
|
"loss": 1.3266,
|
|
"step": 215
|
|
},
|
|
{
|
|
"epoch": 3.208955223880597,
|
|
"eval_loss": 1.60787832736969,
|
|
"eval_runtime": 5.9618,
|
|
"eval_samples_per_second": 79.841,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 215
|
|
},
|
|
{
|
|
"epoch": 3.283582089552239,
|
|
"grad_norm": 1.8919183390013388,
|
|
"learning_rate": 5.0000415857597904e-08,
|
|
"loss": 1.3014,
|
|
"step": 220
|
|
},
|
|
{
|
|
"epoch": 3.283582089552239,
|
|
"eval_loss": 1.6083022356033325,
|
|
"eval_runtime": 5.9802,
|
|
"eval_samples_per_second": 79.596,
|
|
"eval_steps_per_second": 0.669,
|
|
"step": 220
|
|
},
|
|
{
|
|
"epoch": 3.3582089552238807,
|
|
"grad_norm": 1.8867080087928592,
|
|
"learning_rate": 5.0000133662276014e-08,
|
|
"loss": 1.3153,
|
|
"step": 225
|
|
},
|
|
{
|
|
"epoch": 3.3582089552238807,
|
|
"eval_loss": 1.608628273010254,
|
|
"eval_runtime": 5.9587,
|
|
"eval_samples_per_second": 79.883,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 225
|
|
},
|
|
{
|
|
"epoch": 3.4328358208955225,
|
|
"grad_norm": 2.000197758434265,
|
|
"learning_rate": 5.0000037695987735e-08,
|
|
"loss": 1.3192,
|
|
"step": 230
|
|
},
|
|
{
|
|
"epoch": 3.4328358208955225,
|
|
"eval_loss": 1.6089787483215332,
|
|
"eval_runtime": 5.931,
|
|
"eval_samples_per_second": 80.256,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 230
|
|
},
|
|
{
|
|
"epoch": 3.5074626865671643,
|
|
"grad_norm": 1.8236033935396483,
|
|
"learning_rate": 5.0000009015715774e-08,
|
|
"loss": 1.315,
|
|
"step": 235
|
|
},
|
|
{
|
|
"epoch": 3.5074626865671643,
|
|
"eval_loss": 1.6092561483383179,
|
|
"eval_runtime": 5.9999,
|
|
"eval_samples_per_second": 79.335,
|
|
"eval_steps_per_second": 0.667,
|
|
"step": 235
|
|
},
|
|
{
|
|
"epoch": 3.582089552238806,
|
|
"grad_norm": 1.9175089422860454,
|
|
"learning_rate": 5.0000001740409057e-08,
|
|
"loss": 1.3047,
|
|
"step": 240
|
|
},
|
|
{
|
|
"epoch": 3.582089552238806,
|
|
"eval_loss": 1.6093214750289917,
|
|
"eval_runtime": 5.9889,
|
|
"eval_samples_per_second": 79.481,
|
|
"eval_steps_per_second": 0.668,
|
|
"step": 240
|
|
},
|
|
{
|
|
"epoch": 3.656716417910448,
|
|
"grad_norm": 1.8702465201892255,
|
|
"learning_rate": 5.000000025140825e-08,
|
|
"loss": 1.3208,
|
|
"step": 245
|
|
},
|
|
{
|
|
"epoch": 3.656716417910448,
|
|
"eval_loss": 1.609329342842102,
|
|
"eval_runtime": 5.9315,
|
|
"eval_samples_per_second": 80.25,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 245
|
|
},
|
|
{
|
|
"epoch": 3.7313432835820897,
|
|
"grad_norm": 1.9567098448807434,
|
|
"learning_rate": 5.000000002399302e-08,
|
|
"loss": 1.362,
|
|
"step": 250
|
|
},
|
|
{
|
|
"epoch": 3.7313432835820897,
|
|
"eval_loss": 1.609269380569458,
|
|
"eval_runtime": 5.9582,
|
|
"eval_samples_per_second": 79.89,
|
|
"eval_steps_per_second": 0.671,
|
|
"step": 250
|
|
},
|
|
{
|
|
"epoch": 3.8059701492537314,
|
|
"grad_norm": 1.8742551595721921,
|
|
"learning_rate": 5.0000000001205015e-08,
|
|
"loss": 1.3255,
|
|
"step": 255
|
|
},
|
|
{
|
|
"epoch": 3.8059701492537314,
|
|
"eval_loss": 1.609143853187561,
|
|
"eval_runtime": 5.9335,
|
|
"eval_samples_per_second": 80.223,
|
|
"eval_steps_per_second": 0.674,
|
|
"step": 255
|
|
},
|
|
{
|
|
"epoch": 3.8805970149253732,
|
|
"grad_norm": 1.9100964620707728,
|
|
"learning_rate": 5.000000000001954e-08,
|
|
"loss": 1.2941,
|
|
"step": 260
|
|
},
|
|
{
|
|
"epoch": 3.8805970149253732,
|
|
"eval_loss": 1.608902096748352,
|
|
"eval_runtime": 5.9228,
|
|
"eval_samples_per_second": 80.368,
|
|
"eval_steps_per_second": 0.675,
|
|
"step": 260
|
|
},
|
|
{
|
|
"epoch": 3.955223880597015,
|
|
"grad_norm": 1.959684298255238,
|
|
"learning_rate": 5.0000000000000024e-08,
|
|
"loss": 1.3254,
|
|
"step": 265
|
|
},
|
|
{
|
|
"epoch": 3.955223880597015,
|
|
"eval_loss": 1.6086210012435913,
|
|
"eval_runtime": 5.9667,
|
|
"eval_samples_per_second": 79.776,
|
|
"eval_steps_per_second": 0.67,
|
|
"step": 265
|
|
},
|
|
{
|
|
"epoch": 4.0,
|
|
"step": 268,
|
|
"total_flos": 27742804377600.0,
|
|
"train_loss": 1.416380336925165,
|
|
"train_runtime": 8905.9803,
|
|
"train_samples_per_second": 1.923,
|
|
"train_steps_per_second": 0.03
|
|
}
|
|
],
|
|
"logging_steps": 5,
|
|
"max_steps": 268,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 4,
|
|
"save_steps": 5,
|
|
"total_flos": 27742804377600.0,
|
|
"train_batch_size": 8,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|