qwen-compassion-part1 / trainer_state.json
sparrow8i8's picture
add trainer_state.json
7983cbe verified
Raw History Blame Contribute Delete
24.8 kB
{
"best_global_step": 600,
"best_metric": 0.2925775945186615,
"best_model_checkpoint": "/workspace/anchor/qwen-compassion/train/checkpoint-600",
"epoch": 1.0,
"eval_steps": 50,
"global_step": 614,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.008143322475570033,
"grad_norm": 22.442302703857422,
"learning_rate": 2.5806451612903226e-05,
"loss": 3.3662,
"step": 5
},
{
"epoch": 0.016286644951140065,
"grad_norm": 14.803105354309082,
"learning_rate": 5.8064516129032266e-05,
"loss": 3.436,
"step": 10
},
{
"epoch": 0.024429967426710098,
"grad_norm": 9.32910442352295,
"learning_rate": 9.032258064516129e-05,
"loss": 2.2124,
"step": 15
},
{
"epoch": 0.03257328990228013,
"grad_norm": 2.5130510330200195,
"learning_rate": 0.00012258064516129034,
"loss": 1.4505,
"step": 20
},
{
"epoch": 0.04071661237785016,
"grad_norm": 18.144712448120117,
"learning_rate": 0.00015483870967741937,
"loss": 1.1231,
"step": 25
},
{
"epoch": 0.048859934853420196,
"grad_norm": 3.0365991592407227,
"learning_rate": 0.0001870967741935484,
"loss": 0.872,
"step": 30
},
{
"epoch": 0.057003257328990226,
"grad_norm": 2.5769875049591064,
"learning_rate": 0.00019998693330688282,
"loss": 0.8906,
"step": 35
},
{
"epoch": 0.06514657980456026,
"grad_norm": 3.9422550201416016,
"learning_rate": 0.00019990709365929677,
"loss": 0.7195,
"step": 40
},
{
"epoch": 0.0732899022801303,
"grad_norm": 2.395726442337036,
"learning_rate": 0.00019975473152335726,
"loss": 0.6868,
"step": 45
},
{
"epoch": 0.08143322475570032,
"grad_norm": 1.726025104522705,
"learning_rate": 0.0001995299574986306,
"loss": 0.6497,
"step": 50
},
{
"epoch": 0.08143322475570032,
"eval_loss": 0.4577290415763855,
"eval_runtime": 214.9578,
"eval_samples_per_second": 1.172,
"eval_steps_per_second": 0.586,
"step": 50
},
{
"epoch": 0.08957654723127036,
"grad_norm": 1.482474684715271,
"learning_rate": 0.00019923293474842174,
"loss": 0.6646,
"step": 55
},
{
"epoch": 0.09771986970684039,
"grad_norm": 1.236023187637329,
"learning_rate": 0.00019886387888133413,
"loss": 0.5642,
"step": 60
},
{
"epoch": 0.10586319218241043,
"grad_norm": 1.1658198833465576,
"learning_rate": 0.00019842305779475968,
"loss": 0.4603,
"step": 65
},
{
"epoch": 0.11400651465798045,
"grad_norm": 1.2695726156234741,
"learning_rate": 0.0001979107914804122,
"loss": 0.6375,
"step": 70
},
{
"epoch": 0.12214983713355049,
"grad_norm": 1.4793083667755127,
"learning_rate": 0.00019732745179204552,
"loss": 0.4823,
"step": 75
},
{
"epoch": 0.13029315960912052,
"grad_norm": 1.7086373567581177,
"learning_rate": 0.00019667346217552527,
"loss": 0.4646,
"step": 80
},
{
"epoch": 0.13843648208469056,
"grad_norm": 1.3679929971694946,
"learning_rate": 0.00019594929736144976,
"loss": 0.6899,
"step": 85
},
{
"epoch": 0.1465798045602606,
"grad_norm": 1.4010539054870605,
"learning_rate": 0.00019515548302054335,
"loss": 0.5442,
"step": 90
},
{
"epoch": 0.15472312703583063,
"grad_norm": 1.1517243385314941,
"learning_rate": 0.0001942925953820725,
"loss": 0.5184,
"step": 95
},
{
"epoch": 0.16286644951140064,
"grad_norm": 1.058547019958496,
"learning_rate": 0.00019336126081556134,
"loss": 0.6193,
"step": 100
},
{
"epoch": 0.16286644951140064,
"eval_loss": 0.35910215973854065,
"eval_runtime": 120.7216,
"eval_samples_per_second": 2.087,
"eval_steps_per_second": 1.044,
"step": 100
},
{
"epoch": 0.17100977198697068,
"grad_norm": 1.1780120134353638,
"learning_rate": 0.00019236215537611046,
"loss": 0.5177,
"step": 105
},
{
"epoch": 0.1791530944625407,
"grad_norm": 1.2721914052963257,
"learning_rate": 0.00019129600431364897,
"loss": 0.5419,
"step": 110
},
{
"epoch": 0.18729641693811075,
"grad_norm": 1.379878282546997,
"learning_rate": 0.00019016358154647618,
"loss": 0.5167,
"step": 115
},
{
"epoch": 0.19543973941368079,
"grad_norm": 1.2618638277053833,
"learning_rate": 0.00018896570909947475,
"loss": 0.4613,
"step": 120
},
{
"epoch": 0.20358306188925082,
"grad_norm": 1.0391523838043213,
"learning_rate": 0.00018770325650740345,
"loss": 0.4745,
"step": 125
},
{
"epoch": 0.21172638436482086,
"grad_norm": 1.2194364070892334,
"learning_rate": 0.00018637714018370253,
"loss": 0.5386,
"step": 130
},
{
"epoch": 0.21986970684039087,
"grad_norm": 0.7502462267875671,
"learning_rate": 0.00018498832275526988,
"loss": 0.4939,
"step": 135
},
{
"epoch": 0.2280130293159609,
"grad_norm": 0.6979483366012573,
"learning_rate": 0.00018353781236369064,
"loss": 0.4941,
"step": 140
},
{
"epoch": 0.23615635179153094,
"grad_norm": 1.1954421997070312,
"learning_rate": 0.00018202666193342833,
"loss": 0.5449,
"step": 145
},
{
"epoch": 0.24429967426710097,
"grad_norm": 1.3286981582641602,
"learning_rate": 0.00018045596840750723,
"loss": 0.5998,
"step": 150
},
{
"epoch": 0.24429967426710097,
"eval_loss": 0.3380209803581238,
"eval_runtime": 121.0869,
"eval_samples_per_second": 2.081,
"eval_steps_per_second": 1.041,
"step": 150
},
{
"epoch": 0.252442996742671,
"grad_norm": 0.9305353760719299,
"learning_rate": 0.0001788268719512427,
"loss": 0.5237,
"step": 155
},
{
"epoch": 0.26058631921824105,
"grad_norm": 1.387276530265808,
"learning_rate": 0.00017714055512459565,
"loss": 0.4911,
"step": 160
},
{
"epoch": 0.2687296416938111,
"grad_norm": 0.9299679398536682,
"learning_rate": 0.0001753982420237533,
"loss": 0.5527,
"step": 165
},
{
"epoch": 0.2768729641693811,
"grad_norm": 1.2602843046188354,
"learning_rate": 0.00017360119739255852,
"loss": 0.3972,
"step": 170
},
{
"epoch": 0.28501628664495116,
"grad_norm": 0.7769179344177246,
"learning_rate": 0.00017175072570443312,
"loss": 0.5526,
"step": 175
},
{
"epoch": 0.2931596091205212,
"grad_norm": 0.7283039093017578,
"learning_rate": 0.00016984817021546177,
"loss": 0.4304,
"step": 180
},
{
"epoch": 0.30130293159609123,
"grad_norm": 1.4225355386734009,
"learning_rate": 0.000167894911989323,
"loss": 0.4737,
"step": 185
},
{
"epoch": 0.30944625407166126,
"grad_norm": 1.0882941484451294,
"learning_rate": 0.00016589236889477646,
"loss": 0.4035,
"step": 190
},
{
"epoch": 0.31758957654723124,
"grad_norm": 1.1490302085876465,
"learning_rate": 0.00016384199457643262,
"loss": 0.5222,
"step": 195
},
{
"epoch": 0.3257328990228013,
"grad_norm": 0.9430150985717773,
"learning_rate": 0.00016174527739955342,
"loss": 0.6425,
"step": 200
},
{
"epoch": 0.3257328990228013,
"eval_loss": 0.3253232538700104,
"eval_runtime": 120.2962,
"eval_samples_per_second": 2.095,
"eval_steps_per_second": 1.047,
"step": 200
},
{
"epoch": 0.3338762214983713,
"grad_norm": 0.6319507360458374,
"learning_rate": 0.00015960373936964892,
"loss": 0.4485,
"step": 205
},
{
"epoch": 0.34201954397394135,
"grad_norm": 1.1035504341125488,
"learning_rate": 0.0001574189350276545,
"loss": 0.5942,
"step": 210
},
{
"epoch": 0.3501628664495114,
"grad_norm": 1.207892894744873,
"learning_rate": 0.00015519245032149083,
"loss": 0.558,
"step": 215
},
{
"epoch": 0.3583061889250814,
"grad_norm": 0.8965429067611694,
"learning_rate": 0.0001529259014548253,
"loss": 0.478,
"step": 220
},
{
"epoch": 0.36644951140065146,
"grad_norm": 1.1375535726547241,
"learning_rate": 0.00015062093371387097,
"loss": 0.4909,
"step": 225
},
{
"epoch": 0.3745928338762215,
"grad_norm": 1.2835142612457275,
"learning_rate": 0.00014827922027307451,
"loss": 0.4748,
"step": 230
},
{
"epoch": 0.38273615635179153,
"grad_norm": 1.0070925951004028,
"learning_rate": 0.00014590246098055996,
"loss": 0.5095,
"step": 235
},
{
"epoch": 0.39087947882736157,
"grad_norm": 0.7836953401565552,
"learning_rate": 0.00014349238112421024,
"loss": 0.3591,
"step": 240
},
{
"epoch": 0.3990228013029316,
"grad_norm": 0.8382024765014648,
"learning_rate": 0.000141050730179282,
"loss": 0.4465,
"step": 245
},
{
"epoch": 0.40716612377850164,
"grad_norm": 1.106084942817688,
"learning_rate": 0.0001385792805384625,
"loss": 0.6015,
"step": 250
},
{
"epoch": 0.40716612377850164,
"eval_loss": 0.3180280327796936,
"eval_runtime": 120.1946,
"eval_samples_per_second": 2.097,
"eval_steps_per_second": 1.048,
"step": 250
},
{
"epoch": 0.4153094462540717,
"grad_norm": 1.028218150138855,
"learning_rate": 0.00013607982622529133,
"loss": 0.4443,
"step": 255
},
{
"epoch": 0.4234527687296417,
"grad_norm": 3.6740164756774902,
"learning_rate": 0.00013355418159187985,
"loss": 0.5305,
"step": 260
},
{
"epoch": 0.43159609120521175,
"grad_norm": 0.7290945053100586,
"learning_rate": 0.00013100418000187419,
"loss": 0.5237,
"step": 265
},
{
"epoch": 0.43973941368078173,
"grad_norm": 1.0659680366516113,
"learning_rate": 0.0001284316724996181,
"loss": 0.4935,
"step": 270
},
{
"epoch": 0.44788273615635177,
"grad_norm": 1.0143028497695923,
"learning_rate": 0.00012583852646648095,
"loss": 0.6135,
"step": 275
},
{
"epoch": 0.4560260586319218,
"grad_norm": 1.121451735496521,
"learning_rate": 0.00012322662426532708,
"loss": 0.5242,
"step": 280
},
{
"epoch": 0.46416938110749184,
"grad_norm": 0.6729561686515808,
"learning_rate": 0.00012059786187410984,
"loss": 0.3619,
"step": 285
},
{
"epoch": 0.4723127035830619,
"grad_norm": 0.6204145550727844,
"learning_rate": 0.00011795414750958265,
"loss": 0.3198,
"step": 290
},
{
"epoch": 0.4804560260586319,
"grad_norm": 0.9363518953323364,
"learning_rate": 0.00011529740024212565,
"loss": 0.4189,
"step": 295
},
{
"epoch": 0.48859934853420195,
"grad_norm": 1.1184571981430054,
"learning_rate": 0.00011262954860269399,
"loss": 0.587,
"step": 300
},
{
"epoch": 0.48859934853420195,
"eval_loss": 0.311747670173645,
"eval_runtime": 120.3226,
"eval_samples_per_second": 2.094,
"eval_steps_per_second": 1.047,
"step": 300
},
{
"epoch": 0.496742671009772,
"grad_norm": 1.0931415557861328,
"learning_rate": 0.0001099525291828986,
"loss": 0.4254,
"step": 305
},
{
"epoch": 0.504885993485342,
"grad_norm": 1.0900492668151855,
"learning_rate": 0.00010726828522923562,
"loss": 0.4637,
"step": 310
},
{
"epoch": 0.5130293159609121,
"grad_norm": 1.081187129020691,
"learning_rate": 0.00010457876523248518,
"loss": 0.574,
"step": 315
},
{
"epoch": 0.5211726384364821,
"grad_norm": 0.5426526665687561,
"learning_rate": 0.00010188592151330343,
"loss": 0.4139,
"step": 320
},
{
"epoch": 0.5293159609120521,
"grad_norm": 0.6170647144317627,
"learning_rate": 9.919170880503415e-05,
"loss": 0.4787,
"step": 325
},
{
"epoch": 0.5374592833876222,
"grad_norm": 0.9634841084480286,
"learning_rate": 9.649808283476941e-05,
"loss": 0.2533,
"step": 330
},
{
"epoch": 0.5456026058631922,
"grad_norm": 0.6538633108139038,
"learning_rate": 9.38069989036886e-05,
"loss": 0.4913,
"step": 335
},
{
"epoch": 0.5537459283387622,
"grad_norm": 0.6944330334663391,
"learning_rate": 9.112041046770653e-05,
"loss": 0.2606,
"step": 340
},
{
"epoch": 0.5618892508143323,
"grad_norm": 0.9925371408462524,
"learning_rate": 8.844026771946147e-05,
"loss": 0.4495,
"step": 345
},
{
"epoch": 0.5700325732899023,
"grad_norm": 1.0751277208328247,
"learning_rate": 8.57685161726715e-05,
"loss": 0.5536,
"step": 350
},
{
"epoch": 0.5700325732899023,
"eval_loss": 0.3054664731025696,
"eval_runtime": 120.3383,
"eval_samples_per_second": 2.094,
"eval_steps_per_second": 1.047,
"step": 350
},
{
"epoch": 0.5781758957654723,
"grad_norm": 0.8132654428482056,
"learning_rate": 8.310709524988743e-05,
"loss": 0.4582,
"step": 355
},
{
"epoch": 0.5863192182410424,
"grad_norm": 0.9619295001029968,
"learning_rate": 8.045793687466757e-05,
"loss": 0.3663,
"step": 360
},
{
"epoch": 0.5944625407166124,
"grad_norm": 0.9950692057609558,
"learning_rate": 7.782296406919557e-05,
"loss": 0.5359,
"step": 365
},
{
"epoch": 0.6026058631921825,
"grad_norm": 1.0357235670089722,
"learning_rate": 7.520408955836007e-05,
"loss": 0.3589,
"step": 370
},
{
"epoch": 0.6107491856677525,
"grad_norm": 0.6220430731773376,
"learning_rate": 7.260321438130913e-05,
"loss": 0.3868,
"step": 375
},
{
"epoch": 0.6188925081433225,
"grad_norm": 0.7960296869277954,
"learning_rate": 7.002222651148714e-05,
"loss": 0.2924,
"step": 380
},
{
"epoch": 0.6270358306188925,
"grad_norm": 1.227491855621338,
"learning_rate": 6.746299948615631e-05,
"loss": 0.6195,
"step": 385
},
{
"epoch": 0.6351791530944625,
"grad_norm": 0.7982805371284485,
"learning_rate": 6.492739104639727e-05,
"loss": 0.364,
"step": 390
},
{
"epoch": 0.6433224755700325,
"grad_norm": 0.7055819034576416,
"learning_rate": 6.24172417885762e-05,
"loss": 0.4519,
"step": 395
},
{
"epoch": 0.6514657980456026,
"grad_norm": 1.0085915327072144,
"learning_rate": 5.9934373828257106e-05,
"loss": 0.369,
"step": 400
},
{
"epoch": 0.6514657980456026,
"eval_loss": 0.3009154200553894,
"eval_runtime": 120.3254,
"eval_samples_per_second": 2.094,
"eval_steps_per_second": 1.047,
"step": 400
},
{
"epoch": 0.6596091205211726,
"grad_norm": 0.6098869442939758,
"learning_rate": 5.7480589477529545e-05,
"loss": 0.3941,
"step": 405
},
{
"epoch": 0.6677524429967426,
"grad_norm": 0.86444491147995,
"learning_rate": 5.505766993671162e-05,
"loss": 0.4796,
"step": 410
},
{
"epoch": 0.6758957654723127,
"grad_norm": 1.1165945529937744,
"learning_rate": 5.266737400137765e-05,
"loss": 0.3816,
"step": 415
},
{
"epoch": 0.6840390879478827,
"grad_norm": 1.1898753643035889,
"learning_rate": 5.031143678565005e-05,
"loss": 0.3584,
"step": 420
},
{
"epoch": 0.6921824104234527,
"grad_norm": 0.5651485919952393,
"learning_rate": 4.799156846268095e-05,
"loss": 0.3323,
"step": 425
},
{
"epoch": 0.7003257328990228,
"grad_norm": 1.103471279144287,
"learning_rate": 4.57094530232389e-05,
"loss": 0.4991,
"step": 430
},
{
"epoch": 0.7084690553745928,
"grad_norm": 0.9560770988464355,
"learning_rate": 4.346674705330117e-05,
"loss": 0.4969,
"step": 435
},
{
"epoch": 0.7166123778501629,
"grad_norm": 1.0413203239440918,
"learning_rate": 4.1265078531538916e-05,
"loss": 0.3927,
"step": 440
},
{
"epoch": 0.7247557003257329,
"grad_norm": 1.0384825468063354,
"learning_rate": 3.9106045647569e-05,
"loss": 0.4469,
"step": 445
},
{
"epoch": 0.7328990228013029,
"grad_norm": 1.2701938152313232,
"learning_rate": 3.69912156418289e-05,
"loss": 0.5587,
"step": 450
},
{
"epoch": 0.7328990228013029,
"eval_loss": 0.29790911078453064,
"eval_runtime": 120.1996,
"eval_samples_per_second": 2.097,
"eval_steps_per_second": 1.048,
"step": 450
},
{
"epoch": 0.741042345276873,
"grad_norm": 1.088590383529663,
"learning_rate": 3.492212366791831e-05,
"loss": 0.4347,
"step": 455
},
{
"epoch": 0.749185667752443,
"grad_norm": 0.9670878648757935,
"learning_rate": 3.290027167823204e-05,
"loss": 0.4226,
"step": 460
},
{
"epoch": 0.757328990228013,
"grad_norm": 0.9960106015205383,
"learning_rate": 3.092712733369387e-05,
"loss": 0.3849,
"step": 465
},
{
"epoch": 0.7654723127035831,
"grad_norm": 0.6322615742683411,
"learning_rate": 2.9004122938382617e-05,
"loss": 0.3709,
"step": 470
},
{
"epoch": 0.7736156351791531,
"grad_norm": 0.8384042978286743,
"learning_rate": 2.7132654399823444e-05,
"loss": 0.4528,
"step": 475
},
{
"epoch": 0.7817589576547231,
"grad_norm": 0.8728402256965637,
"learning_rate": 2.531408021569982e-05,
"loss": 0.4617,
"step": 480
},
{
"epoch": 0.7899022801302932,
"grad_norm": 0.9127026796340942,
"learning_rate": 2.3549720487720738e-05,
"loss": 0.5415,
"step": 485
},
{
"epoch": 0.7980456026058632,
"grad_norm": 0.9375382661819458,
"learning_rate": 2.184085596336011e-05,
"loss": 0.4591,
"step": 490
},
{
"epoch": 0.8061889250814332,
"grad_norm": 1.062622308731079,
"learning_rate": 2.0188727106162874e-05,
"loss": 0.4813,
"step": 495
},
{
"epoch": 0.8143322475570033,
"grad_norm": 0.8038843870162964,
"learning_rate": 1.8594533195293427e-05,
"loss": 0.4651,
"step": 500
},
{
"epoch": 0.8143322475570033,
"eval_loss": 0.2944928705692291,
"eval_runtime": 120.3478,
"eval_samples_per_second": 2.094,
"eval_steps_per_second": 1.047,
"step": 500
},
{
"epoch": 0.8224755700325733,
"grad_norm": 0.9246629476547241,
"learning_rate": 1.7059431454979824e-05,
"loss": 0.4701,
"step": 505
},
{
"epoch": 0.8306188925081434,
"grad_norm": 1.2579699754714966,
"learning_rate": 1.5584536214485457e-05,
"loss": 0.4114,
"step": 510
},
{
"epoch": 0.8387622149837134,
"grad_norm": 1.0765819549560547,
"learning_rate": 1.4170918099218166e-05,
"loss": 0.4497,
"step": 515
},
{
"epoch": 0.8469055374592834,
"grad_norm": 0.8294546604156494,
"learning_rate": 1.2819603253564205e-05,
"loss": 0.3338,
"step": 520
},
{
"epoch": 0.8550488599348535,
"grad_norm": 0.8488652110099792,
"learning_rate": 1.153157259601062e-05,
"loss": 0.4017,
"step": 525
},
{
"epoch": 0.8631921824104235,
"grad_norm": 0.9699320197105408,
"learning_rate": 1.030776110709718e-05,
"loss": 0.4934,
"step": 530
},
{
"epoch": 0.8713355048859935,
"grad_norm": 0.939552903175354,
"learning_rate": 9.149057150714801e-06,
"loss": 0.4814,
"step": 535
},
{
"epoch": 0.8794788273615635,
"grad_norm": 0.970380961894989,
"learning_rate": 8.056301829242784e-06,
"loss": 0.4135,
"step": 540
},
{
"epoch": 0.8876221498371335,
"grad_norm": 0.5573286414146423,
"learning_rate": 7.030288372993066e-06,
"loss": 0.2724,
"step": 545
},
{
"epoch": 0.8957654723127035,
"grad_norm": 0.9565848112106323,
"learning_rate": 6.0717615644051206e-06,
"loss": 0.4606,
"step": 550
},
{
"epoch": 0.8957654723127035,
"eval_loss": 0.2927910387516022,
"eval_runtime": 120.4502,
"eval_samples_per_second": 2.092,
"eval_steps_per_second": 1.046,
"step": 550
},
{
"epoch": 0.9039087947882736,
"grad_norm": 0.7690985798835754,
"learning_rate": 5.181417197408734e-06,
"loss": 0.4854,
"step": 555
},
{
"epoch": 0.9120521172638436,
"grad_norm": 0.7944011688232422,
"learning_rate": 4.359901572347758e-06,
"loss": 0.425,
"step": 560
},
{
"epoch": 0.9201954397394136,
"grad_norm": 1.0082036256790161,
"learning_rate": 3.607811026831176e-06,
"loss": 0.5035,
"step": 565
},
{
"epoch": 0.9283387622149837,
"grad_norm": 0.9810898303985596,
"learning_rate": 2.9256915028519573e-06,
"loss": 0.3532,
"step": 570
},
{
"epoch": 0.9364820846905537,
"grad_norm": 0.8743959069252014,
"learning_rate": 2.3140381504882737e-06,
"loss": 0.4653,
"step": 575
},
{
"epoch": 0.9446254071661238,
"grad_norm": 0.8244614601135254,
"learning_rate": 1.7732949684743594e-06,
"loss": 0.351,
"step": 580
},
{
"epoch": 0.9527687296416938,
"grad_norm": 1.2245746850967407,
"learning_rate": 1.303854481902067e-06,
"loss": 0.4347,
"step": 585
},
{
"epoch": 0.9609120521172638,
"grad_norm": 0.9963445663452148,
"learning_rate": 9.060574572873237e-07,
"loss": 0.4403,
"step": 590
},
{
"epoch": 0.9690553745928339,
"grad_norm": 1.018866777420044,
"learning_rate": 5.801926552078563e-07,
"loss": 0.4273,
"step": 595
},
{
"epoch": 0.9771986970684039,
"grad_norm": 1.0799201726913452,
"learning_rate": 3.264966206921294e-07,
"loss": 0.4367,
"step": 600
},
{
"epoch": 0.9771986970684039,
"eval_loss": 0.2925775945186615,
"eval_runtime": 120.683,
"eval_samples_per_second": 2.088,
"eval_steps_per_second": 1.044,
"step": 600
},
{
"epoch": 0.9853420195439739,
"grad_norm": 1.1354663372039795,
"learning_rate": 1.451535115114866e-07,
"loss": 0.478,
"step": 605
},
{
"epoch": 0.993485342019544,
"grad_norm": 0.8019553422927856,
"learning_rate": 3.629496450011649e-08,
"loss": 0.3683,
"step": 610
}
],
"logging_steps": 5,
"max_steps": 614,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 2,
"early_stopping_threshold": 0.001
},
"attributes": {
"early_stopping_patience_counter": 1
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 2.508719122176676e+18,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}