codellama_instruct_pt_text2sql / trainer_state.json
tmnam20's picture
End of training
8344920
{
"best_metric": 0.01498666312545538,
"best_model_checkpoint": "./text2sql/codellama_instruct_pt_text2sql/checkpoint-44000",
"epoch": 4.9995126366358,
"eval_steps": 2000,
"global_step": 44880,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 1.1139992573338283e-07,
"loss": 1.2191,
"step": 1
},
{
"epoch": 0.11,
"learning_rate": 0.00011139992573338284,
"loss": 0.2684,
"step": 1000
},
{
"epoch": 0.22,
"learning_rate": 0.00022279985146676567,
"loss": 0.0693,
"step": 2000
},
{
"epoch": 0.22,
"eval_loss": 0.05887996032834053,
"eval_runtime": 171.127,
"eval_samples_per_second": 11.687,
"eval_steps_per_second": 1.461,
"step": 2000
},
{
"epoch": 0.33,
"learning_rate": 0.00029781686301467276,
"loss": 0.0565,
"step": 3000
},
{
"epoch": 0.45,
"learning_rate": 0.0002907056676227274,
"loss": 0.047,
"step": 4000
},
{
"epoch": 0.45,
"eval_loss": 0.03957173973321915,
"eval_runtime": 171.3682,
"eval_samples_per_second": 11.671,
"eval_steps_per_second": 1.459,
"step": 4000
},
{
"epoch": 0.56,
"learning_rate": 0.00028359447223078195,
"loss": 0.0403,
"step": 5000
},
{
"epoch": 0.67,
"learning_rate": 0.0002764832768388366,
"loss": 0.0364,
"step": 6000
},
{
"epoch": 0.67,
"eval_loss": 0.030724667012691498,
"eval_runtime": 170.9405,
"eval_samples_per_second": 11.7,
"eval_steps_per_second": 1.462,
"step": 6000
},
{
"epoch": 0.78,
"learning_rate": 0.0002693720814468912,
"loss": 0.0341,
"step": 7000
},
{
"epoch": 0.89,
"learning_rate": 0.00026226088605494583,
"loss": 0.0311,
"step": 8000
},
{
"epoch": 0.89,
"eval_loss": 0.027806701138615608,
"eval_runtime": 171.064,
"eval_samples_per_second": 11.692,
"eval_steps_per_second": 1.461,
"step": 8000
},
{
"epoch": 1.0,
"learning_rate": 0.00025514969066300046,
"loss": 0.0289,
"step": 9000
},
{
"epoch": 1.11,
"learning_rate": 0.00024803849527105503,
"loss": 0.0251,
"step": 10000
},
{
"epoch": 1.11,
"eval_loss": 0.024082325398921967,
"eval_runtime": 171.4689,
"eval_samples_per_second": 11.664,
"eval_steps_per_second": 1.458,
"step": 10000
},
{
"epoch": 1.23,
"learning_rate": 0.00024092729987910966,
"loss": 0.0242,
"step": 11000
},
{
"epoch": 1.34,
"learning_rate": 0.00023381610448716428,
"loss": 0.0243,
"step": 12000
},
{
"epoch": 1.34,
"eval_loss": 0.02279273048043251,
"eval_runtime": 171.2147,
"eval_samples_per_second": 11.681,
"eval_steps_per_second": 1.46,
"step": 12000
},
{
"epoch": 1.45,
"learning_rate": 0.00022670490909521888,
"loss": 0.0219,
"step": 13000
},
{
"epoch": 1.56,
"learning_rate": 0.0002195937137032735,
"loss": 0.0227,
"step": 14000
},
{
"epoch": 1.56,
"eval_loss": 0.022252563387155533,
"eval_runtime": 171.1834,
"eval_samples_per_second": 11.683,
"eval_steps_per_second": 1.46,
"step": 14000
},
{
"epoch": 1.67,
"learning_rate": 0.00021248251831132813,
"loss": 0.0218,
"step": 15000
},
{
"epoch": 1.78,
"learning_rate": 0.00020537132291938273,
"loss": 0.0212,
"step": 16000
},
{
"epoch": 1.78,
"eval_loss": 0.020056582987308502,
"eval_runtime": 170.9191,
"eval_samples_per_second": 11.701,
"eval_steps_per_second": 1.463,
"step": 16000
},
{
"epoch": 1.89,
"learning_rate": 0.00019826012752743733,
"loss": 0.0208,
"step": 17000
},
{
"epoch": 2.01,
"learning_rate": 0.00019114893213549196,
"loss": 0.0202,
"step": 18000
},
{
"epoch": 2.01,
"eval_loss": 0.01821763999760151,
"eval_runtime": 171.4736,
"eval_samples_per_second": 11.664,
"eval_steps_per_second": 1.458,
"step": 18000
},
{
"epoch": 2.12,
"learning_rate": 0.00018403773674354658,
"loss": 0.0161,
"step": 19000
},
{
"epoch": 2.23,
"learning_rate": 0.0001769265413516012,
"loss": 0.016,
"step": 20000
},
{
"epoch": 2.23,
"eval_loss": 0.018352985382080078,
"eval_runtime": 171.2111,
"eval_samples_per_second": 11.681,
"eval_steps_per_second": 1.46,
"step": 20000
},
{
"epoch": 2.34,
"learning_rate": 0.00016981534595965578,
"loss": 0.0155,
"step": 21000
},
{
"epoch": 2.45,
"learning_rate": 0.0001627041505677104,
"loss": 0.0156,
"step": 22000
},
{
"epoch": 2.45,
"eval_loss": 0.017868032678961754,
"eval_runtime": 171.2047,
"eval_samples_per_second": 11.682,
"eval_steps_per_second": 1.46,
"step": 22000
},
{
"epoch": 2.56,
"learning_rate": 0.00015559295517576503,
"loss": 0.0155,
"step": 23000
},
{
"epoch": 2.67,
"learning_rate": 0.00014848175978381966,
"loss": 0.015,
"step": 24000
},
{
"epoch": 2.67,
"eval_loss": 0.017283763736486435,
"eval_runtime": 171.6023,
"eval_samples_per_second": 11.655,
"eval_steps_per_second": 1.457,
"step": 24000
},
{
"epoch": 2.78,
"learning_rate": 0.00014137056439187426,
"loss": 0.0153,
"step": 25000
},
{
"epoch": 2.9,
"learning_rate": 0.00013425936899992888,
"loss": 0.0147,
"step": 26000
},
{
"epoch": 2.9,
"eval_loss": 0.016522083431482315,
"eval_runtime": 171.211,
"eval_samples_per_second": 11.681,
"eval_steps_per_second": 1.46,
"step": 26000
},
{
"epoch": 3.01,
"learning_rate": 0.00012714817360798348,
"loss": 0.0137,
"step": 27000
},
{
"epoch": 3.12,
"learning_rate": 0.00012003697821603811,
"loss": 0.0112,
"step": 28000
},
{
"epoch": 3.12,
"eval_loss": 0.016529470682144165,
"eval_runtime": 170.9168,
"eval_samples_per_second": 11.702,
"eval_steps_per_second": 1.463,
"step": 28000
},
{
"epoch": 3.23,
"learning_rate": 0.00011292578282409272,
"loss": 0.0112,
"step": 29000
},
{
"epoch": 3.34,
"learning_rate": 0.00010581458743214733,
"loss": 0.0109,
"step": 30000
},
{
"epoch": 3.34,
"eval_loss": 0.016144245862960815,
"eval_runtime": 171.5226,
"eval_samples_per_second": 11.66,
"eval_steps_per_second": 1.458,
"step": 30000
},
{
"epoch": 3.45,
"learning_rate": 9.870339204020194e-05,
"loss": 0.0111,
"step": 31000
},
{
"epoch": 3.56,
"learning_rate": 9.159219664825657e-05,
"loss": 0.0109,
"step": 32000
},
{
"epoch": 3.56,
"eval_loss": 0.015489176847040653,
"eval_runtime": 171.4873,
"eval_samples_per_second": 11.663,
"eval_steps_per_second": 1.458,
"step": 32000
},
{
"epoch": 3.68,
"learning_rate": 8.448100125631117e-05,
"loss": 0.0111,
"step": 33000
},
{
"epoch": 3.79,
"learning_rate": 7.73698058643658e-05,
"loss": 0.0105,
"step": 34000
},
{
"epoch": 3.79,
"eval_loss": 0.015172351151704788,
"eval_runtime": 171.4626,
"eval_samples_per_second": 11.664,
"eval_steps_per_second": 1.458,
"step": 34000
},
{
"epoch": 3.9,
"learning_rate": 7.025861047242041e-05,
"loss": 0.0107,
"step": 35000
},
{
"epoch": 4.01,
"learning_rate": 6.314741508047502e-05,
"loss": 0.0104,
"step": 36000
},
{
"epoch": 4.01,
"eval_loss": 0.014995447359979153,
"eval_runtime": 171.1146,
"eval_samples_per_second": 11.688,
"eval_steps_per_second": 1.461,
"step": 36000
},
{
"epoch": 4.12,
"learning_rate": 5.603621968852963e-05,
"loss": 0.0081,
"step": 37000
},
{
"epoch": 4.23,
"learning_rate": 4.8925024296584245e-05,
"loss": 0.0077,
"step": 38000
},
{
"epoch": 4.23,
"eval_loss": 0.015779005363583565,
"eval_runtime": 171.0935,
"eval_samples_per_second": 11.69,
"eval_steps_per_second": 1.461,
"step": 38000
},
{
"epoch": 4.34,
"learning_rate": 4.181382890463887e-05,
"loss": 0.0079,
"step": 39000
},
{
"epoch": 4.46,
"learning_rate": 3.4702633512693484e-05,
"loss": 0.0078,
"step": 40000
},
{
"epoch": 4.46,
"eval_loss": 0.015086671337485313,
"eval_runtime": 171.2287,
"eval_samples_per_second": 11.68,
"eval_steps_per_second": 1.46,
"step": 40000
},
{
"epoch": 4.57,
"learning_rate": 2.7591438120748093e-05,
"loss": 0.0079,
"step": 41000
},
{
"epoch": 4.68,
"learning_rate": 2.0480242728802712e-05,
"loss": 0.0076,
"step": 42000
},
{
"epoch": 4.68,
"eval_loss": 0.015026736073195934,
"eval_runtime": 170.9263,
"eval_samples_per_second": 11.701,
"eval_steps_per_second": 1.463,
"step": 42000
},
{
"epoch": 4.79,
"learning_rate": 1.3369047336857326e-05,
"loss": 0.0075,
"step": 43000
},
{
"epoch": 4.9,
"learning_rate": 6.257851944911939e-06,
"loss": 0.0077,
"step": 44000
},
{
"epoch": 4.9,
"eval_loss": 0.01498666312545538,
"eval_runtime": 171.2767,
"eval_samples_per_second": 11.677,
"eval_steps_per_second": 1.46,
"step": 44000
},
{
"epoch": 5.0,
"step": 44880,
"total_flos": 2.184131254639421e+19,
"train_loss": 0.0249389049336447,
"train_runtime": 262166.9813,
"train_samples_per_second": 5.479,
"train_steps_per_second": 0.171
}
],
"logging_steps": 1000,
"max_steps": 44880,
"num_train_epochs": 5,
"save_steps": 2000,
"total_flos": 2.184131254639421e+19,
"trial_name": null,
"trial_params": null
}