Spindle-LLM / trainer_state.json
DMID23's picture
Upload 7 files
bac07e4 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"eval_steps": 500,
"global_step": 735,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.040878896269800714,
"grad_norm": 6.032278537750244,
"learning_rate": 0.00019755102040816327,
"loss": 2.6293,
"step": 10
},
{
"epoch": 0.08175779253960143,
"grad_norm": 3.237109899520874,
"learning_rate": 0.0001948299319727891,
"loss": 2.008,
"step": 20
},
{
"epoch": 0.12263668880940215,
"grad_norm": 2.7581236362457275,
"learning_rate": 0.000192108843537415,
"loss": 1.855,
"step": 30
},
{
"epoch": 0.16351558507920286,
"grad_norm": 3.083601474761963,
"learning_rate": 0.00018938775510204083,
"loss": 1.7576,
"step": 40
},
{
"epoch": 0.20439448134900357,
"grad_norm": 2.7807652950286865,
"learning_rate": 0.0001866666666666667,
"loss": 1.7434,
"step": 50
},
{
"epoch": 0.2452733776188043,
"grad_norm": 2.6213436126708984,
"learning_rate": 0.00018394557823129252,
"loss": 1.682,
"step": 60
},
{
"epoch": 0.286152273888605,
"grad_norm": 2.6907846927642822,
"learning_rate": 0.00018122448979591838,
"loss": 1.6749,
"step": 70
},
{
"epoch": 0.3270311701584057,
"grad_norm": 5.332385540008545,
"learning_rate": 0.00017850340136054421,
"loss": 1.5744,
"step": 80
},
{
"epoch": 0.36791006642820645,
"grad_norm": 3.23760986328125,
"learning_rate": 0.00017578231292517008,
"loss": 1.5338,
"step": 90
},
{
"epoch": 0.40878896269800713,
"grad_norm": 3.479809522628784,
"learning_rate": 0.00017306122448979594,
"loss": 1.4143,
"step": 100
},
{
"epoch": 0.44966785896780787,
"grad_norm": 2.7317419052124023,
"learning_rate": 0.0001703401360544218,
"loss": 1.4603,
"step": 110
},
{
"epoch": 0.4905467552376086,
"grad_norm": 2.8279190063476562,
"learning_rate": 0.00016761904761904763,
"loss": 1.5169,
"step": 120
},
{
"epoch": 0.5314256515074093,
"grad_norm": 3.096827983856201,
"learning_rate": 0.0001648979591836735,
"loss": 1.4396,
"step": 130
},
{
"epoch": 0.57230454777721,
"grad_norm": 2.9124717712402344,
"learning_rate": 0.00016217687074829932,
"loss": 1.3824,
"step": 140
},
{
"epoch": 0.6131834440470108,
"grad_norm": 3.073765754699707,
"learning_rate": 0.00015945578231292519,
"loss": 1.3974,
"step": 150
},
{
"epoch": 0.6540623403168114,
"grad_norm": 2.5229296684265137,
"learning_rate": 0.00015673469387755102,
"loss": 1.4211,
"step": 160
},
{
"epoch": 0.6949412365866121,
"grad_norm": 2.8301663398742676,
"learning_rate": 0.00015401360544217688,
"loss": 1.3567,
"step": 170
},
{
"epoch": 0.7358201328564129,
"grad_norm": 2.6760520935058594,
"learning_rate": 0.00015129251700680274,
"loss": 1.3552,
"step": 180
},
{
"epoch": 0.7766990291262136,
"grad_norm": 2.741240978240967,
"learning_rate": 0.00014857142857142857,
"loss": 1.408,
"step": 190
},
{
"epoch": 0.8175779253960143,
"grad_norm": 2.028970956802368,
"learning_rate": 0.00014585034013605443,
"loss": 1.3497,
"step": 200
},
{
"epoch": 0.858456821665815,
"grad_norm": 2.6956677436828613,
"learning_rate": 0.00014312925170068027,
"loss": 1.3592,
"step": 210
},
{
"epoch": 0.8993357179356157,
"grad_norm": 2.448338270187378,
"learning_rate": 0.00014040816326530613,
"loss": 1.3309,
"step": 220
},
{
"epoch": 0.9402146142054164,
"grad_norm": 2.267707347869873,
"learning_rate": 0.00013768707482993196,
"loss": 1.329,
"step": 230
},
{
"epoch": 0.9810935104752172,
"grad_norm": 2.4610676765441895,
"learning_rate": 0.00013496598639455782,
"loss": 1.3194,
"step": 240
},
{
"epoch": 1.0204394481349004,
"grad_norm": 2.539577007293701,
"learning_rate": 0.00013224489795918368,
"loss": 1.2862,
"step": 250
},
{
"epoch": 1.0613183444047012,
"grad_norm": 2.377258777618408,
"learning_rate": 0.00012952380952380954,
"loss": 1.2052,
"step": 260
},
{
"epoch": 1.1021972406745018,
"grad_norm": 2.4516634941101074,
"learning_rate": 0.00012680272108843538,
"loss": 1.1859,
"step": 270
},
{
"epoch": 1.1430761369443025,
"grad_norm": 3.4123401641845703,
"learning_rate": 0.00012408163265306124,
"loss": 1.2547,
"step": 280
},
{
"epoch": 1.1839550332141031,
"grad_norm": 2.9211533069610596,
"learning_rate": 0.00012136054421768707,
"loss": 1.187,
"step": 290
},
{
"epoch": 1.224833929483904,
"grad_norm": 2.8444571495056152,
"learning_rate": 0.00011863945578231292,
"loss": 1.1871,
"step": 300
},
{
"epoch": 1.2657128257537047,
"grad_norm": 2.8576762676239014,
"learning_rate": 0.00011591836734693877,
"loss": 1.2033,
"step": 310
},
{
"epoch": 1.3065917220235055,
"grad_norm": 2.8612818717956543,
"learning_rate": 0.00011319727891156464,
"loss": 1.1889,
"step": 320
},
{
"epoch": 1.347470618293306,
"grad_norm": 2.666062355041504,
"learning_rate": 0.00011047619047619049,
"loss": 1.1122,
"step": 330
},
{
"epoch": 1.3883495145631068,
"grad_norm": 2.7399775981903076,
"learning_rate": 0.00010775510204081634,
"loss": 1.2343,
"step": 340
},
{
"epoch": 1.4292284108329074,
"grad_norm": 2.89231538772583,
"learning_rate": 0.00010503401360544218,
"loss": 1.2481,
"step": 350
},
{
"epoch": 1.4701073071027082,
"grad_norm": 2.9003829956054688,
"learning_rate": 0.00010231292517006803,
"loss": 1.1994,
"step": 360
},
{
"epoch": 1.510986203372509,
"grad_norm": 3.2669005393981934,
"learning_rate": 9.959183673469388e-05,
"loss": 1.1804,
"step": 370
},
{
"epoch": 1.5518650996423098,
"grad_norm": 3.1723055839538574,
"learning_rate": 9.687074829931974e-05,
"loss": 1.1427,
"step": 380
},
{
"epoch": 1.5927439959121104,
"grad_norm": 3.0546045303344727,
"learning_rate": 9.414965986394559e-05,
"loss": 1.1616,
"step": 390
},
{
"epoch": 1.633622892181911,
"grad_norm": 3.060725450515747,
"learning_rate": 9.142857142857143e-05,
"loss": 1.2045,
"step": 400
},
{
"epoch": 1.6745017884517117,
"grad_norm": 2.8336000442504883,
"learning_rate": 8.870748299319729e-05,
"loss": 1.1922,
"step": 410
},
{
"epoch": 1.7153806847215125,
"grad_norm": 2.888324499130249,
"learning_rate": 8.598639455782314e-05,
"loss": 1.1646,
"step": 420
},
{
"epoch": 1.7562595809913133,
"grad_norm": 2.559555768966675,
"learning_rate": 8.326530612244899e-05,
"loss": 1.2204,
"step": 430
},
{
"epoch": 1.797138477261114,
"grad_norm": 3.291555643081665,
"learning_rate": 8.054421768707483e-05,
"loss": 1.1561,
"step": 440
},
{
"epoch": 1.8380173735309147,
"grad_norm": 2.7448337078094482,
"learning_rate": 7.782312925170068e-05,
"loss": 1.1783,
"step": 450
},
{
"epoch": 1.8788962698007152,
"grad_norm": 3.201742649078369,
"learning_rate": 7.510204081632653e-05,
"loss": 1.1499,
"step": 460
},
{
"epoch": 1.919775166070516,
"grad_norm": 3.3023853302001953,
"learning_rate": 7.238095238095238e-05,
"loss": 1.1929,
"step": 470
},
{
"epoch": 1.9606540623403168,
"grad_norm": 2.9392805099487305,
"learning_rate": 6.965986394557822e-05,
"loss": 1.2178,
"step": 480
},
{
"epoch": 2.0,
"grad_norm": 3.641012668609619,
"learning_rate": 6.693877551020408e-05,
"loss": 1.1721,
"step": 490
},
{
"epoch": 2.040878896269801,
"grad_norm": 3.4774110317230225,
"learning_rate": 6.421768707482993e-05,
"loss": 1.0634,
"step": 500
},
{
"epoch": 2.0817577925396016,
"grad_norm": 3.280494213104248,
"learning_rate": 6.149659863945578e-05,
"loss": 1.0549,
"step": 510
},
{
"epoch": 2.1226366888094024,
"grad_norm": 3.7747931480407715,
"learning_rate": 5.877551020408164e-05,
"loss": 1.0331,
"step": 520
},
{
"epoch": 2.1635155850792027,
"grad_norm": 3.1277899742126465,
"learning_rate": 5.6054421768707486e-05,
"loss": 1.0384,
"step": 530
},
{
"epoch": 2.2043944813490035,
"grad_norm": 3.6484415531158447,
"learning_rate": 5.333333333333333e-05,
"loss": 1.0136,
"step": 540
},
{
"epoch": 2.2452733776188043,
"grad_norm": 3.5595340728759766,
"learning_rate": 5.061224489795918e-05,
"loss": 1.0227,
"step": 550
},
{
"epoch": 2.286152273888605,
"grad_norm": 3.5124669075012207,
"learning_rate": 4.7891156462585034e-05,
"loss": 1.0109,
"step": 560
},
{
"epoch": 2.327031170158406,
"grad_norm": 3.5096631050109863,
"learning_rate": 4.517006802721089e-05,
"loss": 1.0131,
"step": 570
},
{
"epoch": 2.3679100664282062,
"grad_norm": 3.6057543754577637,
"learning_rate": 4.2448979591836735e-05,
"loss": 1.0791,
"step": 580
},
{
"epoch": 2.408788962698007,
"grad_norm": 3.254915714263916,
"learning_rate": 3.972789115646259e-05,
"loss": 0.9997,
"step": 590
},
{
"epoch": 2.449667858967808,
"grad_norm": 4.024539947509766,
"learning_rate": 3.7006802721088437e-05,
"loss": 1.0364,
"step": 600
},
{
"epoch": 2.4905467552376086,
"grad_norm": 3.6602048873901367,
"learning_rate": 3.428571428571429e-05,
"loss": 0.9388,
"step": 610
},
{
"epoch": 2.5314256515074094,
"grad_norm": 3.5694384574890137,
"learning_rate": 3.156462585034014e-05,
"loss": 1.0256,
"step": 620
},
{
"epoch": 2.57230454777721,
"grad_norm": 2.8324692249298096,
"learning_rate": 2.8843537414965988e-05,
"loss": 1.0396,
"step": 630
},
{
"epoch": 2.613183444047011,
"grad_norm": 3.545496702194214,
"learning_rate": 2.612244897959184e-05,
"loss": 1.0569,
"step": 640
},
{
"epoch": 2.6540623403168113,
"grad_norm": 3.602918863296509,
"learning_rate": 2.3401360544217686e-05,
"loss": 1.0485,
"step": 650
},
{
"epoch": 2.694941236586612,
"grad_norm": 3.526660442352295,
"learning_rate": 2.0680272108843536e-05,
"loss": 1.0541,
"step": 660
},
{
"epoch": 2.735820132856413,
"grad_norm": 3.0326685905456543,
"learning_rate": 1.7959183673469387e-05,
"loss": 0.9885,
"step": 670
},
{
"epoch": 2.7766990291262137,
"grad_norm": 3.41005277633667,
"learning_rate": 1.5238095238095241e-05,
"loss": 1.0277,
"step": 680
},
{
"epoch": 2.817577925396014,
"grad_norm": 3.461618423461914,
"learning_rate": 1.251700680272109e-05,
"loss": 1.0553,
"step": 690
},
{
"epoch": 2.858456821665815,
"grad_norm": 3.59061861038208,
"learning_rate": 9.795918367346939e-06,
"loss": 1.0436,
"step": 700
},
{
"epoch": 2.8993357179356156,
"grad_norm": 3.4869349002838135,
"learning_rate": 7.07482993197279e-06,
"loss": 1.0493,
"step": 710
},
{
"epoch": 2.9402146142054164,
"grad_norm": 3.2588963508605957,
"learning_rate": 4.353741496598639e-06,
"loss": 1.0311,
"step": 720
},
{
"epoch": 2.981093510475217,
"grad_norm": 3.391941785812378,
"learning_rate": 1.63265306122449e-06,
"loss": 0.9652,
"step": 730
}
],
"logging_steps": 10,
"max_steps": 735,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"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": 2.5661456203854643e+17,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}