RepairLLaMA-IR1-OR3 / trainer_state.json
Sen Fang
Up
e466bb4
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 10.0,
"global_step": 1310,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02,
"learning_rate": 0.0004999281136632892,
"loss": 0.3104,
"step": 10
},
{
"epoch": 0.03,
"learning_rate": 0.0004997124959943201,
"loss": 0.148,
"step": 20
},
{
"epoch": 0.05,
"learning_rate": 0.0004993532709928075,
"loss": 0.1311,
"step": 30
},
{
"epoch": 0.06,
"learning_rate": 0.0004988506452457066,
"loss": 0.1266,
"step": 40
},
{
"epoch": 0.08,
"learning_rate": 0.0004982049078084071,
"loss": 0.1169,
"step": 50
},
{
"epoch": 0.09,
"learning_rate": 0.0004974164300384998,
"loss": 0.1142,
"step": 60
},
{
"epoch": 0.11,
"learning_rate": 0.0004964856653822122,
"loss": 0.1128,
"step": 70
},
{
"epoch": 0.12,
"learning_rate": 0.0004954131491136361,
"loss": 0.1103,
"step": 80
},
{
"epoch": 0.14,
"learning_rate": 0.0004941994980268967,
"loss": 0.1188,
"step": 90
},
{
"epoch": 0.15,
"learning_rate": 0.000492845410081439,
"loss": 0.1028,
"step": 100
},
{
"epoch": 0.17,
"learning_rate": 0.0004913516640006391,
"loss": 0.1076,
"step": 110
},
{
"epoch": 0.18,
"learning_rate": 0.0004897191188239667,
"loss": 0.1114,
"step": 120
},
{
"epoch": 0.2,
"learning_rate": 0.00048794871341296,
"loss": 0.1084,
"step": 130
},
{
"epoch": 0.21,
"learning_rate": 0.00048604146591129483,
"loss": 0.1107,
"step": 140
},
{
"epoch": 0.23,
"learning_rate": 0.00048399847315926,
"loss": 0.1062,
"step": 150
},
{
"epoch": 0.24,
"learning_rate": 0.00048182091006297446,
"loss": 0.1021,
"step": 160
},
{
"epoch": 0.26,
"learning_rate": 0.00047951002891870987,
"loss": 0.102,
"step": 170
},
{
"epoch": 0.27,
"learning_rate": 0.00047706715869270635,
"loss": 0.098,
"step": 180
},
{
"epoch": 0.29,
"learning_rate": 0.000474493704256897,
"loss": 0.103,
"step": 190
},
{
"epoch": 0.31,
"learning_rate": 0.0004717911455809782,
"loss": 0.1111,
"step": 200
},
{
"epoch": 0.32,
"learning_rate": 0.0004689610368812938,
"loss": 0.1033,
"step": 210
},
{
"epoch": 0.34,
"learning_rate": 0.0004660050057270191,
"loss": 0.1018,
"step": 220
},
{
"epoch": 0.35,
"learning_rate": 0.0004629247521041611,
"loss": 0.1108,
"step": 230
},
{
"epoch": 0.37,
"learning_rate": 0.0004597220474379125,
"loss": 0.1084,
"step": 240
},
{
"epoch": 0.38,
"learning_rate": 0.0004563987335739216,
"loss": 0.1084,
"step": 250
},
{
"epoch": 0.4,
"learning_rate": 0.00045295672171906365,
"loss": 0.1156,
"step": 260
},
{
"epoch": 0.41,
"learning_rate": 0.00044939799134232397,
"loss": 0.1076,
"step": 270
},
{
"epoch": 0.43,
"learning_rate": 0.0004457245890364235,
"loss": 0.1077,
"step": 280
},
{
"epoch": 0.44,
"learning_rate": 0.0004419386273408428,
"loss": 0.1034,
"step": 290
},
{
"epoch": 0.46,
"learning_rate": 0.0004380422835269193,
"loss": 0.1045,
"step": 300
},
{
"epoch": 0.47,
"learning_rate": 0.00043403779834572,
"loss": 0.1017,
"step": 310
},
{
"epoch": 0.49,
"learning_rate": 0.00042992747473940553,
"loss": 0.1092,
"step": 320
},
{
"epoch": 0.5,
"learning_rate": 0.00042571367651682995,
"loss": 0.1042,
"step": 330
},
{
"epoch": 0.52,
"learning_rate": 0.00042139882699413617,
"loss": 0.0961,
"step": 340
},
{
"epoch": 0.53,
"learning_rate": 0.00041698540760112926,
"loss": 0.1003,
"step": 350
},
{
"epoch": 0.55,
"learning_rate": 0.00041247595645422955,
"loss": 0.1027,
"step": 360
},
{
"epoch": 0.56,
"learning_rate": 0.0004078730668968252,
"loss": 0.0969,
"step": 370
},
{
"epoch": 0.58,
"learning_rate": 0.00040317938600786487,
"loss": 0.1028,
"step": 380
},
{
"epoch": 0.6,
"learning_rate": 0.00039839761307954675,
"loss": 0.1093,
"step": 390
},
{
"epoch": 0.61,
"learning_rate": 0.0003935304980649813,
"loss": 0.1115,
"step": 400
},
{
"epoch": 0.63,
"learning_rate": 0.0003885808399967186,
"loss": 0.0995,
"step": 410
},
{
"epoch": 0.64,
"learning_rate": 0.0003835514853770505,
"loss": 0.0961,
"step": 420
},
{
"epoch": 0.66,
"learning_rate": 0.00037844532654101405,
"loss": 0.0966,
"step": 430
},
{
"epoch": 0.67,
"learning_rate": 0.0003732652999930364,
"loss": 0.0992,
"step": 440
},
{
"epoch": 0.69,
"learning_rate": 0.00036801438471817827,
"loss": 0.1008,
"step": 450
},
{
"epoch": 0.7,
"learning_rate": 0.00036269560046894763,
"loss": 0.1018,
"step": 460
},
{
"epoch": 0.72,
"learning_rate": 0.0003573120060286679,
"loss": 0.0995,
"step": 470
},
{
"epoch": 0.73,
"learning_rate": 0.00035186669745240024,
"loss": 0.1009,
"step": 480
},
{
"epoch": 0.75,
"learning_rate": 0.0003463628062864312,
"loss": 0.0998,
"step": 490
},
{
"epoch": 0.76,
"learning_rate": 0.00034080349776734924,
"loss": 0.1043,
"step": 500
},
{
"epoch": 0.78,
"learning_rate": 0.00033519196900174726,
"loss": 0.1006,
"step": 510
},
{
"epoch": 0.79,
"learning_rate": 0.0003295314471275954,
"loss": 0.0975,
"step": 520
},
{
"epoch": 0.81,
"learning_rate": 0.00032382518745834515,
"loss": 0.0987,
"step": 530
},
{
"epoch": 0.82,
"learning_rate": 0.00031807647161082795,
"loss": 0.0996,
"step": 540
},
{
"epoch": 0.84,
"learning_rate": 0.0003122886056180284,
"loss": 0.1042,
"step": 550
},
{
"epoch": 0.85,
"learning_rate": 0.00030646491802781517,
"loss": 0.1044,
"step": 560
},
{
"epoch": 0.87,
"learning_rate": 0.00030060875798872436,
"loss": 0.1033,
"step": 570
},
{
"epoch": 0.89,
"learning_rate": 0.00029472349332389523,
"loss": 0.0983,
"step": 580
},
{
"epoch": 0.9,
"learning_rate": 0.0002888125085942664,
"loss": 0.1014,
"step": 590
},
{
"epoch": 0.92,
"learning_rate": 0.00028287920315214646,
"loss": 0.0949,
"step": 600
},
{
"epoch": 0.93,
"learning_rate": 0.0002769269891862778,
"loss": 0.1037,
"step": 610
},
{
"epoch": 0.95,
"learning_rate": 0.0002709592897595191,
"loss": 0.0971,
"step": 620
},
{
"epoch": 0.96,
"learning_rate": 0.0002649795368402735,
"loss": 0.0943,
"step": 630
},
{
"epoch": 0.98,
"learning_rate": 0.00025899116932879537,
"loss": 0.102,
"step": 640
},
{
"epoch": 0.99,
"learning_rate": 0.0002529976310795108,
"loss": 0.0926,
"step": 650
},
{
"epoch": 1.01,
"learning_rate": 0.0002470023689204893,
"loss": 0.0945,
"step": 660
},
{
"epoch": 1.02,
"learning_rate": 0.00024100883067120475,
"loss": 0.0964,
"step": 670
},
{
"epoch": 1.04,
"learning_rate": 0.00023502046315972656,
"loss": 0.0889,
"step": 680
},
{
"epoch": 1.05,
"learning_rate": 0.0002290407102404809,
"loss": 0.0916,
"step": 690
},
{
"epoch": 1.07,
"learning_rate": 0.00022307301081372224,
"loss": 0.0928,
"step": 700
},
{
"epoch": 1.08,
"learning_rate": 0.00021712079684785363,
"loss": 0.0937,
"step": 710
},
{
"epoch": 1.1,
"learning_rate": 0.0002111874914057336,
"loss": 0.0893,
"step": 720
},
{
"epoch": 1.11,
"learning_rate": 0.00020527650667610475,
"loss": 0.0923,
"step": 730
},
{
"epoch": 1.13,
"learning_rate": 0.0001993912420112756,
"loss": 0.092,
"step": 740
},
{
"epoch": 1.15,
"learning_rate": 0.00019353508197218492,
"loss": 0.0949,
"step": 750
},
{
"epoch": 1.16,
"learning_rate": 0.00018771139438197168,
"loss": 0.0909,
"step": 760
},
{
"epoch": 1.18,
"learning_rate": 0.00018192352838917208,
"loss": 0.0912,
"step": 770
},
{
"epoch": 1.19,
"learning_rate": 0.0001761748125416549,
"loss": 0.0915,
"step": 780
},
{
"epoch": 1.21,
"learning_rate": 0.00017046855287240463,
"loss": 0.0947,
"step": 790
},
{
"epoch": 1.22,
"learning_rate": 0.00016480803099825278,
"loss": 0.0891,
"step": 800
},
{
"epoch": 1.24,
"learning_rate": 0.0001591965022326507,
"loss": 0.0979,
"step": 810
},
{
"epoch": 1.25,
"learning_rate": 0.00015363719371356882,
"loss": 0.0881,
"step": 820
},
{
"epoch": 1.27,
"learning_rate": 0.00014813330254759985,
"loss": 0.0895,
"step": 830
},
{
"epoch": 1.28,
"learning_rate": 0.0001426879939713322,
"loss": 0.0931,
"step": 840
},
{
"epoch": 1.3,
"learning_rate": 0.00013730439953105243,
"loss": 0.0928,
"step": 850
},
{
"epoch": 1.31,
"learning_rate": 0.00013198561528182182,
"loss": 0.0978,
"step": 860
},
{
"epoch": 1.33,
"learning_rate": 0.00012673470000696365,
"loss": 0.0916,
"step": 870
},
{
"epoch": 1.34,
"learning_rate": 0.00012155467345898602,
"loss": 0.0885,
"step": 880
},
{
"epoch": 1.36,
"learning_rate": 0.00011644851462294955,
"loss": 0.0959,
"step": 890
},
{
"epoch": 1.37,
"learning_rate": 0.00011141916000328151,
"loss": 0.0929,
"step": 900
},
{
"epoch": 1.39,
"learning_rate": 0.0001064695019350187,
"loss": 0.0924,
"step": 910
},
{
"epoch": 1.4,
"learning_rate": 0.00010160238692045331,
"loss": 0.0926,
"step": 920
},
{
"epoch": 1.42,
"learning_rate": 9.682061399213524e-05,
"loss": 0.0976,
"step": 930
},
{
"epoch": 1.44,
"learning_rate": 9.21269331031748e-05,
"loss": 0.0958,
"step": 940
},
{
"epoch": 1.45,
"learning_rate": 8.752404354577052e-05,
"loss": 0.0921,
"step": 950
},
{
"epoch": 1.47,
"learning_rate": 8.301459239887074e-05,
"loss": 0.0973,
"step": 960
},
{
"epoch": 1.48,
"learning_rate": 7.860117300586384e-05,
"loss": 0.0957,
"step": 970
},
{
"epoch": 1.5,
"learning_rate": 7.428632348317005e-05,
"loss": 0.099,
"step": 980
},
{
"epoch": 1.51,
"learning_rate": 7.007252526059446e-05,
"loss": 0.0934,
"step": 990
},
{
"epoch": 1.53,
"learning_rate": 6.596220165428002e-05,
"loss": 0.0925,
"step": 1000
},
{
"epoch": 1.54,
"learning_rate": 6.195771647308074e-05,
"loss": 0.0937,
"step": 1010
},
{
"epoch": 1.56,
"learning_rate": 5.8061372659157316e-05,
"loss": 0.0979,
"step": 1020
},
{
"epoch": 1.57,
"learning_rate": 5.427541096357647e-05,
"loss": 0.099,
"step": 1030
},
{
"epoch": 1.59,
"learning_rate": 5.060200865767606e-05,
"loss": 0.0915,
"step": 1040
},
{
"epoch": 1.6,
"learning_rate": 4.704327828093641e-05,
"loss": 0.0931,
"step": 1050
},
{
"epoch": 1.62,
"learning_rate": 4.360126642607842e-05,
"loss": 0.095,
"step": 1060
},
{
"epoch": 1.63,
"learning_rate": 4.027795256208749e-05,
"loss": 0.095,
"step": 1070
},
{
"epoch": 1.65,
"learning_rate": 3.707524789583891e-05,
"loss": 0.095,
"step": 1080
},
{
"epoch": 1.66,
"learning_rate": 3.3994994272980944e-05,
"loss": 0.0898,
"step": 1090
},
{
"epoch": 1.68,
"learning_rate": 3.1038963118706244e-05,
"loss": 0.0901,
"step": 1100
},
{
"epoch": 1.69,
"learning_rate": 2.8208854419021823e-05,
"loss": 0.0902,
"step": 1110
},
{
"epoch": 1.71,
"learning_rate": 2.5506295743103092e-05,
"loss": 0.0945,
"step": 1120
},
{
"epoch": 1.73,
"learning_rate": 2.2932841307293644e-05,
"loss": 0.0904,
"step": 1130
},
{
"epoch": 1.74,
"learning_rate": 2.0489971081290192e-05,
"loss": 0.0895,
"step": 1140
},
{
"epoch": 1.76,
"learning_rate": 1.817908993702555e-05,
"loss": 0.0936,
"step": 1150
},
{
"epoch": 1.77,
"learning_rate": 1.600152684074005e-05,
"loss": 0.0927,
"step": 1160
},
{
"epoch": 1.79,
"learning_rate": 1.3958534088705205e-05,
"loss": 0.0947,
"step": 1170
},
{
"epoch": 1.8,
"learning_rate": 1.205128658704005e-05,
"loss": 0.0993,
"step": 1180
},
{
"epoch": 1.82,
"learning_rate": 1.028088117603332e-05,
"loss": 0.0894,
"step": 1190
},
{
"epoch": 1.83,
"learning_rate": 8.648335999360934e-06,
"loss": 0.0909,
"step": 1200
},
{
"epoch": 1.85,
"learning_rate": 7.154589918561022e-06,
"loss": 0.0886,
"step": 1210
},
{
"epoch": 1.86,
"learning_rate": 5.800501973103362e-06,
"loss": 0.0949,
"step": 1220
},
{
"epoch": 1.88,
"learning_rate": 4.586850886363874e-06,
"loss": 0.0958,
"step": 1230
},
{
"epoch": 1.89,
"learning_rate": 3.5143346177878568e-06,
"loss": 0.0939,
"step": 1240
},
{
"epoch": 1.91,
"learning_rate": 2.583569961500276e-06,
"loss": 0.0932,
"step": 1250
},
{
"epoch": 1.92,
"learning_rate": 1.7950921915928787e-06,
"loss": 0.0972,
"step": 1260
},
{
"epoch": 1.94,
"learning_rate": 1.149354754293397e-06,
"loss": 0.088,
"step": 1270
},
{
"epoch": 1.95,
"learning_rate": 6.467290071925646e-07,
"loss": 0.0955,
"step": 1280
},
{
"epoch": 1.97,
"learning_rate": 2.8750400567992274e-07,
"loss": 0.0879,
"step": 1290
},
{
"epoch": 1.98,
"learning_rate": 7.188633671079136e-08,
"loss": 0.0985,
"step": 1300
},
{
"epoch": 2.0,
"learning_rate": 0.0,
"loss": 0.091,
"step": 1310
},
{
"epoch": 2.0,
"step": 1310,
"total_flos": 3.149409890779267e+18,
"train_loss": 0.10076319215861895,
"train_runtime": 21644.7687,
"train_samples_per_second": 3.873,
"train_steps_per_second": 0.061
}
],
"logging_steps": 10,
"max_steps": 1310,
"num_train_epochs": 2,
"save_steps": 150,
"total_flos": 3.149409890779267e+18,
"trial_name": null,
"trial_params": null
}