andre-3000 / trainer_state.json
AlekseyKorshuk's picture
huggingartists
b4c3898
raw
history blame
13.2 kB
{
"best_metric": 3.5885772705078125,
"best_model_checkpoint": "output/andre-3000/checkpoint-500",
"epoch": 5.0,
"global_step": 500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05,
"learning_rate": 0.00013635542016482644,
"loss": 3.842,
"step": 5
},
{
"epoch": 0.1,
"learning_rate": 0.00013384247701784754,
"loss": 4.137,
"step": 10
},
{
"epoch": 0.15,
"learning_rate": 0.00012972304755932203,
"loss": 3.733,
"step": 15
},
{
"epoch": 0.2,
"learning_rate": 0.0001240985658141214,
"loss": 3.9016,
"step": 20
},
{
"epoch": 0.25,
"learning_rate": 0.00011710752518939715,
"loss": 3.9056,
"step": 25
},
{
"epoch": 0.3,
"learning_rate": 0.00010892206830726365,
"loss": 3.4673,
"step": 30
},
{
"epoch": 0.35,
"learning_rate": 9.974374828213291e-05,
"loss": 3.4192,
"step": 35
},
{
"epoch": 0.4,
"learning_rate": 8.97985658141214e-05,
"loss": 3.5188,
"step": 40
},
{
"epoch": 0.45,
"learning_rate": 7.933140430175984e-05,
"loss": 3.6982,
"step": 45
},
{
"epoch": 0.5,
"learning_rate": 6.86e-05,
"loss": 3.6325,
"step": 50
},
{
"epoch": 0.55,
"learning_rate": 5.786859569824015e-05,
"loss": 3.5144,
"step": 55
},
{
"epoch": 0.6,
"learning_rate": 4.740143418587861e-05,
"loss": 3.6288,
"step": 60
},
{
"epoch": 0.65,
"learning_rate": 3.745625171786709e-05,
"loss": 3.5069,
"step": 65
},
{
"epoch": 0.7,
"learning_rate": 2.8277931692736352e-05,
"loss": 3.7103,
"step": 70
},
{
"epoch": 0.75,
"learning_rate": 2.0092474810602843e-05,
"loss": 3.5622,
"step": 75
},
{
"epoch": 0.8,
"learning_rate": 1.3101434185878613e-05,
"loss": 3.5587,
"step": 80
},
{
"epoch": 0.85,
"learning_rate": 7.476952440677969e-06,
"loss": 3.5987,
"step": 85
},
{
"epoch": 0.9,
"learning_rate": 3.357522982152468e-06,
"loss": 3.4558,
"step": 90
},
{
"epoch": 0.95,
"learning_rate": 8.445798351735566e-07,
"loss": 3.528,
"step": 95
},
{
"epoch": 1.0,
"learning_rate": 0.0,
"loss": 3.5143,
"step": 100
},
{
"epoch": 1.0,
"eval_loss": 3.681246519088745,
"eval_runtime": 2.8995,
"eval_samples_per_second": 46.215,
"eval_steps_per_second": 5.863,
"step": 100
},
{
"epoch": 1.05,
"learning_rate": 8.445798351735489e-07,
"loss": 3.5508,
"step": 105
},
{
"epoch": 1.1,
"learning_rate": 3.357522982152468e-06,
"loss": 3.6326,
"step": 110
},
{
"epoch": 1.15,
"learning_rate": 7.4769524406779465e-06,
"loss": 3.3899,
"step": 115
},
{
"epoch": 1.2,
"learning_rate": 1.3101434185878598e-05,
"loss": 3.3572,
"step": 120
},
{
"epoch": 1.25,
"learning_rate": 2.009247481060283e-05,
"loss": 3.5502,
"step": 125
},
{
"epoch": 1.3,
"learning_rate": 2.8277931692736335e-05,
"loss": 3.3994,
"step": 130
},
{
"epoch": 1.35,
"learning_rate": 3.745625171786708e-05,
"loss": 3.2566,
"step": 135
},
{
"epoch": 1.4,
"learning_rate": 4.74014341858786e-05,
"loss": 3.503,
"step": 140
},
{
"epoch": 1.45,
"learning_rate": 5.786859569824015e-05,
"loss": 3.0786,
"step": 145
},
{
"epoch": 1.5,
"learning_rate": 6.859999999999999e-05,
"loss": 3.2728,
"step": 150
},
{
"epoch": 1.55,
"learning_rate": 7.933140430175983e-05,
"loss": 3.1862,
"step": 155
},
{
"epoch": 1.6,
"learning_rate": 8.979856581412138e-05,
"loss": 3.2618,
"step": 160
},
{
"epoch": 1.65,
"learning_rate": 9.974374828213291e-05,
"loss": 3.4498,
"step": 165
},
{
"epoch": 1.7,
"learning_rate": 0.00010892206830726364,
"loss": 3.1053,
"step": 170
},
{
"epoch": 1.75,
"learning_rate": 0.00011710752518939715,
"loss": 3.3994,
"step": 175
},
{
"epoch": 1.8,
"learning_rate": 0.0001240985658141214,
"loss": 3.3686,
"step": 180
},
{
"epoch": 1.85,
"learning_rate": 0.00012972304755932203,
"loss": 3.4802,
"step": 185
},
{
"epoch": 1.9,
"learning_rate": 0.00013384247701784754,
"loss": 3.4496,
"step": 190
},
{
"epoch": 1.95,
"learning_rate": 0.00013635542016482644,
"loss": 3.341,
"step": 195
},
{
"epoch": 2.0,
"learning_rate": 0.0001372,
"loss": 3.2332,
"step": 200
},
{
"epoch": 2.0,
"eval_loss": 3.6474390029907227,
"eval_runtime": 3.0601,
"eval_samples_per_second": 43.789,
"eval_steps_per_second": 5.555,
"step": 200
},
{
"epoch": 2.05,
"learning_rate": 0.00013635542016482644,
"loss": 3.2607,
"step": 205
},
{
"epoch": 2.1,
"learning_rate": 0.00013384247701784754,
"loss": 3.3,
"step": 210
},
{
"epoch": 2.15,
"learning_rate": 0.00012972304755932206,
"loss": 3.3456,
"step": 215
},
{
"epoch": 2.2,
"learning_rate": 0.00012409856581412136,
"loss": 3.0185,
"step": 220
},
{
"epoch": 2.25,
"learning_rate": 0.00011710752518939717,
"loss": 2.9552,
"step": 225
},
{
"epoch": 2.3,
"learning_rate": 0.00010892206830726372,
"loss": 3.0603,
"step": 230
},
{
"epoch": 2.35,
"learning_rate": 9.974374828213292e-05,
"loss": 3.2515,
"step": 235
},
{
"epoch": 2.4,
"learning_rate": 8.979856581412141e-05,
"loss": 3.1474,
"step": 240
},
{
"epoch": 2.45,
"learning_rate": 7.93314043017598e-05,
"loss": 3.1997,
"step": 245
},
{
"epoch": 2.5,
"learning_rate": 6.860000000000001e-05,
"loss": 3.2995,
"step": 250
},
{
"epoch": 2.55,
"learning_rate": 5.786859569824025e-05,
"loss": 3.0081,
"step": 255
},
{
"epoch": 2.6,
"learning_rate": 4.7401434185878625e-05,
"loss": 3.0519,
"step": 260
},
{
"epoch": 2.65,
"learning_rate": 3.745625171786717e-05,
"loss": 3.1474,
"step": 265
},
{
"epoch": 2.7,
"learning_rate": 2.827793169273636e-05,
"loss": 3.0318,
"step": 270
},
{
"epoch": 2.75,
"learning_rate": 2.0092474810602897e-05,
"loss": 3.0418,
"step": 275
},
{
"epoch": 2.8,
"learning_rate": 1.310143418587862e-05,
"loss": 3.2108,
"step": 280
},
{
"epoch": 2.85,
"learning_rate": 7.4769524406779465e-06,
"loss": 2.8269,
"step": 285
},
{
"epoch": 2.9,
"learning_rate": 3.3575229821524754e-06,
"loss": 3.0926,
"step": 290
},
{
"epoch": 2.95,
"learning_rate": 8.445798351735489e-07,
"loss": 3.1617,
"step": 295
},
{
"epoch": 3.0,
"learning_rate": 0.0,
"loss": 2.8832,
"step": 300
},
{
"epoch": 3.0,
"eval_loss": 3.5949366092681885,
"eval_runtime": 3.0805,
"eval_samples_per_second": 43.5,
"eval_steps_per_second": 5.519,
"step": 300
},
{
"epoch": 3.05,
"learning_rate": 8.445798351735413e-07,
"loss": 2.8947,
"step": 305
},
{
"epoch": 3.1,
"learning_rate": 3.35752298215246e-06,
"loss": 2.8681,
"step": 310
},
{
"epoch": 3.15,
"learning_rate": 7.4769524406779245e-06,
"loss": 2.925,
"step": 315
},
{
"epoch": 3.2,
"learning_rate": 1.310143418587859e-05,
"loss": 3.0341,
"step": 320
},
{
"epoch": 3.25,
"learning_rate": 2.0092474810602867e-05,
"loss": 2.923,
"step": 325
},
{
"epoch": 3.3,
"learning_rate": 2.827793169273632e-05,
"loss": 3.1311,
"step": 330
},
{
"epoch": 3.35,
"learning_rate": 3.7456251717867126e-05,
"loss": 2.9954,
"step": 335
},
{
"epoch": 3.4,
"learning_rate": 4.740143418587858e-05,
"loss": 2.9511,
"step": 340
},
{
"epoch": 3.45,
"learning_rate": 5.7868595698240195e-05,
"loss": 2.9837,
"step": 345
},
{
"epoch": 3.5,
"learning_rate": 6.859999999999997e-05,
"loss": 3.0341,
"step": 350
},
{
"epoch": 3.55,
"learning_rate": 7.933140430175976e-05,
"loss": 2.9587,
"step": 355
},
{
"epoch": 3.6,
"learning_rate": 8.979856581412137e-05,
"loss": 2.9158,
"step": 360
},
{
"epoch": 3.65,
"learning_rate": 9.974374828213283e-05,
"loss": 2.8452,
"step": 365
},
{
"epoch": 3.7,
"learning_rate": 0.00010892206830726364,
"loss": 2.9248,
"step": 370
},
{
"epoch": 3.75,
"learning_rate": 0.0001171075251893971,
"loss": 2.9796,
"step": 375
},
{
"epoch": 3.8,
"learning_rate": 0.00012409856581412136,
"loss": 3.0723,
"step": 380
},
{
"epoch": 3.85,
"learning_rate": 0.00012972304755932206,
"loss": 2.7475,
"step": 385
},
{
"epoch": 3.9,
"learning_rate": 0.00013384247701784751,
"loss": 2.9859,
"step": 390
},
{
"epoch": 3.95,
"learning_rate": 0.00013635542016482644,
"loss": 2.8605,
"step": 395
},
{
"epoch": 4.0,
"learning_rate": 0.0001372,
"loss": 2.8721,
"step": 400
},
{
"epoch": 4.0,
"eval_loss": 3.6242599487304688,
"eval_runtime": 3.0946,
"eval_samples_per_second": 43.301,
"eval_steps_per_second": 5.493,
"step": 400
},
{
"epoch": 4.05,
"learning_rate": 0.00013635542016482647,
"loss": 3.0586,
"step": 405
},
{
"epoch": 4.1,
"learning_rate": 0.0001338424770178476,
"loss": 2.5264,
"step": 410
},
{
"epoch": 4.15,
"learning_rate": 0.00012972304755932203,
"loss": 2.591,
"step": 415
},
{
"epoch": 4.2,
"learning_rate": 0.00012409856581412142,
"loss": 2.7559,
"step": 420
},
{
"epoch": 4.25,
"learning_rate": 0.00011710752518939714,
"loss": 2.6444,
"step": 425
},
{
"epoch": 4.3,
"learning_rate": 0.00010892206830726369,
"loss": 3.0661,
"step": 430
},
{
"epoch": 4.35,
"learning_rate": 9.9743748282133e-05,
"loss": 2.639,
"step": 435
},
{
"epoch": 4.4,
"learning_rate": 8.97985658141213e-05,
"loss": 2.8619,
"step": 440
},
{
"epoch": 4.45,
"learning_rate": 7.933140430175981e-05,
"loss": 2.6417,
"step": 445
},
{
"epoch": 4.5,
"learning_rate": 6.860000000000003e-05,
"loss": 2.7156,
"step": 450
},
{
"epoch": 4.55,
"learning_rate": 5.786859569824026e-05,
"loss": 2.9429,
"step": 455
},
{
"epoch": 4.6,
"learning_rate": 4.740143418587876e-05,
"loss": 2.7116,
"step": 460
},
{
"epoch": 4.65,
"learning_rate": 3.745625171786707e-05,
"loss": 2.6837,
"step": 465
},
{
"epoch": 4.7,
"learning_rate": 2.8277931692736372e-05,
"loss": 2.8497,
"step": 470
},
{
"epoch": 4.75,
"learning_rate": 2.0092474810602914e-05,
"loss": 2.713,
"step": 475
},
{
"epoch": 4.8,
"learning_rate": 1.3101434185878628e-05,
"loss": 2.7103,
"step": 480
},
{
"epoch": 4.85,
"learning_rate": 7.4769524406780075e-06,
"loss": 2.6698,
"step": 485
},
{
"epoch": 4.9,
"learning_rate": 3.3575229821524373e-06,
"loss": 2.8479,
"step": 490
},
{
"epoch": 4.95,
"learning_rate": 8.445798351735489e-07,
"loss": 2.7422,
"step": 495
},
{
"epoch": 5.0,
"learning_rate": 0.0,
"loss": 2.8229,
"step": 500
},
{
"epoch": 5.0,
"eval_loss": 3.5885772705078125,
"eval_runtime": 3.0956,
"eval_samples_per_second": 43.288,
"eval_steps_per_second": 5.492,
"step": 500
}
],
"max_steps": 1000,
"num_train_epochs": 10,
"total_flos": 521277603840000.0,
"trial_name": null,
"trial_params": null
}