| { |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 2.995488721804511, |
| "eval_steps": 500, |
| "global_step": 996, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.03007518796992481, |
| "grad_norm": 5.823914796640075, |
| "learning_rate": 5e-06, |
| "loss": 0.801, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.06015037593984962, |
| "grad_norm": 2.3399969132192253, |
| "learning_rate": 5e-06, |
| "loss": 0.7172, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.09022556390977443, |
| "grad_norm": 0.8783578572732146, |
| "learning_rate": 5e-06, |
| "loss": 0.6923, |
| "step": 30 |
| }, |
| { |
| "epoch": 0.12030075187969924, |
| "grad_norm": 1.7232554735132826, |
| "learning_rate": 5e-06, |
| "loss": 0.6834, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.15037593984962405, |
| "grad_norm": 1.6494035164725938, |
| "learning_rate": 5e-06, |
| "loss": 0.6655, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.18045112781954886, |
| "grad_norm": 0.9942658093552336, |
| "learning_rate": 5e-06, |
| "loss": 0.6568, |
| "step": 60 |
| }, |
| { |
| "epoch": 0.21052631578947367, |
| "grad_norm": 0.8609069727317351, |
| "learning_rate": 5e-06, |
| "loss": 0.6548, |
| "step": 70 |
| }, |
| { |
| "epoch": 0.24060150375939848, |
| "grad_norm": 0.8740060444495542, |
| "learning_rate": 5e-06, |
| "loss": 0.6454, |
| "step": 80 |
| }, |
| { |
| "epoch": 0.2706766917293233, |
| "grad_norm": 1.0114201009544495, |
| "learning_rate": 5e-06, |
| "loss": 0.6331, |
| "step": 90 |
| }, |
| { |
| "epoch": 0.3007518796992481, |
| "grad_norm": 0.782353210012559, |
| "learning_rate": 5e-06, |
| "loss": 0.6267, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.3308270676691729, |
| "grad_norm": 0.6061308240718153, |
| "learning_rate": 5e-06, |
| "loss": 0.6305, |
| "step": 110 |
| }, |
| { |
| "epoch": 0.3609022556390977, |
| "grad_norm": 0.663886451339362, |
| "learning_rate": 5e-06, |
| "loss": 0.6246, |
| "step": 120 |
| }, |
| { |
| "epoch": 0.39097744360902253, |
| "grad_norm": 0.8509285444900951, |
| "learning_rate": 5e-06, |
| "loss": 0.6305, |
| "step": 130 |
| }, |
| { |
| "epoch": 0.42105263157894735, |
| "grad_norm": 0.650634026198435, |
| "learning_rate": 5e-06, |
| "loss": 0.6291, |
| "step": 140 |
| }, |
| { |
| "epoch": 0.45112781954887216, |
| "grad_norm": 0.7050446482120244, |
| "learning_rate": 5e-06, |
| "loss": 0.6279, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.48120300751879697, |
| "grad_norm": 0.5260864128397903, |
| "learning_rate": 5e-06, |
| "loss": 0.6167, |
| "step": 160 |
| }, |
| { |
| "epoch": 0.5112781954887218, |
| "grad_norm": 0.5159636362174167, |
| "learning_rate": 5e-06, |
| "loss": 0.6189, |
| "step": 170 |
| }, |
| { |
| "epoch": 0.5413533834586466, |
| "grad_norm": 0.576027072578746, |
| "learning_rate": 5e-06, |
| "loss": 0.6096, |
| "step": 180 |
| }, |
| { |
| "epoch": 0.5714285714285714, |
| "grad_norm": 0.6020714354457843, |
| "learning_rate": 5e-06, |
| "loss": 0.6207, |
| "step": 190 |
| }, |
| { |
| "epoch": 0.6015037593984962, |
| "grad_norm": 0.8904744763001984, |
| "learning_rate": 5e-06, |
| "loss": 0.6143, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.631578947368421, |
| "grad_norm": 0.5645606661034147, |
| "learning_rate": 5e-06, |
| "loss": 0.619, |
| "step": 210 |
| }, |
| { |
| "epoch": 0.6616541353383458, |
| "grad_norm": 0.7456282489942, |
| "learning_rate": 5e-06, |
| "loss": 0.6229, |
| "step": 220 |
| }, |
| { |
| "epoch": 0.6917293233082706, |
| "grad_norm": 0.47671003637879944, |
| "learning_rate": 5e-06, |
| "loss": 0.6141, |
| "step": 230 |
| }, |
| { |
| "epoch": 0.7218045112781954, |
| "grad_norm": 0.5712162016443497, |
| "learning_rate": 5e-06, |
| "loss": 0.6148, |
| "step": 240 |
| }, |
| { |
| "epoch": 0.7518796992481203, |
| "grad_norm": 0.5787527818544967, |
| "learning_rate": 5e-06, |
| "loss": 0.6112, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.7819548872180451, |
| "grad_norm": 0.7727895820276254, |
| "learning_rate": 5e-06, |
| "loss": 0.6124, |
| "step": 260 |
| }, |
| { |
| "epoch": 0.8120300751879699, |
| "grad_norm": 0.6627350412484229, |
| "learning_rate": 5e-06, |
| "loss": 0.6101, |
| "step": 270 |
| }, |
| { |
| "epoch": 0.8421052631578947, |
| "grad_norm": 0.49307733921025576, |
| "learning_rate": 5e-06, |
| "loss": 0.6101, |
| "step": 280 |
| }, |
| { |
| "epoch": 0.8721804511278195, |
| "grad_norm": 0.5299740752648865, |
| "learning_rate": 5e-06, |
| "loss": 0.6074, |
| "step": 290 |
| }, |
| { |
| "epoch": 0.9022556390977443, |
| "grad_norm": 0.9821666854504033, |
| "learning_rate": 5e-06, |
| "loss": 0.6035, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.9323308270676691, |
| "grad_norm": 0.5537120584191855, |
| "learning_rate": 5e-06, |
| "loss": 0.6037, |
| "step": 310 |
| }, |
| { |
| "epoch": 0.9624060150375939, |
| "grad_norm": 0.5346047989400851, |
| "learning_rate": 5e-06, |
| "loss": 0.6128, |
| "step": 320 |
| }, |
| { |
| "epoch": 0.9924812030075187, |
| "grad_norm": 0.5177704737274331, |
| "learning_rate": 5e-06, |
| "loss": 0.6043, |
| "step": 330 |
| }, |
| { |
| "epoch": 0.9984962406015038, |
| "eval_loss": 0.6096732020378113, |
| "eval_runtime": 116.9563, |
| "eval_samples_per_second": 76.593, |
| "eval_steps_per_second": 0.599, |
| "step": 332 |
| }, |
| { |
| "epoch": 1.0225563909774436, |
| "grad_norm": 0.4996873605367385, |
| "learning_rate": 5e-06, |
| "loss": 0.58, |
| "step": 340 |
| }, |
| { |
| "epoch": 1.0526315789473684, |
| "grad_norm": 0.59979801353788, |
| "learning_rate": 5e-06, |
| "loss": 0.5596, |
| "step": 350 |
| }, |
| { |
| "epoch": 1.0827067669172932, |
| "grad_norm": 0.645810347625552, |
| "learning_rate": 5e-06, |
| "loss": 0.5591, |
| "step": 360 |
| }, |
| { |
| "epoch": 1.112781954887218, |
| "grad_norm": 0.6113621459091085, |
| "learning_rate": 5e-06, |
| "loss": 0.5553, |
| "step": 370 |
| }, |
| { |
| "epoch": 1.1428571428571428, |
| "grad_norm": 0.6400931476209645, |
| "learning_rate": 5e-06, |
| "loss": 0.5564, |
| "step": 380 |
| }, |
| { |
| "epoch": 1.1729323308270676, |
| "grad_norm": 0.6192317007821239, |
| "learning_rate": 5e-06, |
| "loss": 0.5605, |
| "step": 390 |
| }, |
| { |
| "epoch": 1.2030075187969924, |
| "grad_norm": 0.5237078678145406, |
| "learning_rate": 5e-06, |
| "loss": 0.5519, |
| "step": 400 |
| }, |
| { |
| "epoch": 1.2330827067669172, |
| "grad_norm": 0.5140051147329998, |
| "learning_rate": 5e-06, |
| "loss": 0.5574, |
| "step": 410 |
| }, |
| { |
| "epoch": 1.263157894736842, |
| "grad_norm": 0.6369697944320184, |
| "learning_rate": 5e-06, |
| "loss": 0.5541, |
| "step": 420 |
| }, |
| { |
| "epoch": 1.2932330827067668, |
| "grad_norm": 0.5985062376730702, |
| "learning_rate": 5e-06, |
| "loss": 0.5582, |
| "step": 430 |
| }, |
| { |
| "epoch": 1.3233082706766917, |
| "grad_norm": 0.48850272868134653, |
| "learning_rate": 5e-06, |
| "loss": 0.5532, |
| "step": 440 |
| }, |
| { |
| "epoch": 1.3533834586466165, |
| "grad_norm": 0.5920840023317582, |
| "learning_rate": 5e-06, |
| "loss": 0.5571, |
| "step": 450 |
| }, |
| { |
| "epoch": 1.3834586466165413, |
| "grad_norm": 0.5287134171137652, |
| "learning_rate": 5e-06, |
| "loss": 0.5487, |
| "step": 460 |
| }, |
| { |
| "epoch": 1.413533834586466, |
| "grad_norm": 0.48569117162380887, |
| "learning_rate": 5e-06, |
| "loss": 0.5586, |
| "step": 470 |
| }, |
| { |
| "epoch": 1.443609022556391, |
| "grad_norm": 0.5524909003831754, |
| "learning_rate": 5e-06, |
| "loss": 0.5528, |
| "step": 480 |
| }, |
| { |
| "epoch": 1.4736842105263157, |
| "grad_norm": 0.45528161895298735, |
| "learning_rate": 5e-06, |
| "loss": 0.5624, |
| "step": 490 |
| }, |
| { |
| "epoch": 1.5037593984962405, |
| "grad_norm": 0.5654083015816137, |
| "learning_rate": 5e-06, |
| "loss": 0.5565, |
| "step": 500 |
| }, |
| { |
| "epoch": 1.5338345864661656, |
| "grad_norm": 0.5087876058685868, |
| "learning_rate": 5e-06, |
| "loss": 0.5587, |
| "step": 510 |
| }, |
| { |
| "epoch": 1.5639097744360901, |
| "grad_norm": 0.4290062049121276, |
| "learning_rate": 5e-06, |
| "loss": 0.5493, |
| "step": 520 |
| }, |
| { |
| "epoch": 1.5939849624060152, |
| "grad_norm": 0.6235091434876753, |
| "learning_rate": 5e-06, |
| "loss": 0.5603, |
| "step": 530 |
| }, |
| { |
| "epoch": 1.6240601503759398, |
| "grad_norm": 0.5314512113272476, |
| "learning_rate": 5e-06, |
| "loss": 0.5564, |
| "step": 540 |
| }, |
| { |
| "epoch": 1.6541353383458648, |
| "grad_norm": 0.5049026603516721, |
| "learning_rate": 5e-06, |
| "loss": 0.5605, |
| "step": 550 |
| }, |
| { |
| "epoch": 1.6842105263157894, |
| "grad_norm": 0.5650921119799542, |
| "learning_rate": 5e-06, |
| "loss": 0.5601, |
| "step": 560 |
| }, |
| { |
| "epoch": 1.7142857142857144, |
| "grad_norm": 0.5193160484708012, |
| "learning_rate": 5e-06, |
| "loss": 0.5554, |
| "step": 570 |
| }, |
| { |
| "epoch": 1.744360902255639, |
| "grad_norm": 0.5213313242863221, |
| "learning_rate": 5e-06, |
| "loss": 0.5645, |
| "step": 580 |
| }, |
| { |
| "epoch": 1.774436090225564, |
| "grad_norm": 0.6327600595776153, |
| "learning_rate": 5e-06, |
| "loss": 0.5536, |
| "step": 590 |
| }, |
| { |
| "epoch": 1.8045112781954886, |
| "grad_norm": 0.5461472827380875, |
| "learning_rate": 5e-06, |
| "loss": 0.5625, |
| "step": 600 |
| }, |
| { |
| "epoch": 1.8345864661654137, |
| "grad_norm": 0.6541180795154581, |
| "learning_rate": 5e-06, |
| "loss": 0.554, |
| "step": 610 |
| }, |
| { |
| "epoch": 1.8646616541353382, |
| "grad_norm": 0.5572692340571848, |
| "learning_rate": 5e-06, |
| "loss": 0.5529, |
| "step": 620 |
| }, |
| { |
| "epoch": 1.8947368421052633, |
| "grad_norm": 0.5724927689201437, |
| "learning_rate": 5e-06, |
| "loss": 0.547, |
| "step": 630 |
| }, |
| { |
| "epoch": 1.9248120300751879, |
| "grad_norm": 0.48899030621990486, |
| "learning_rate": 5e-06, |
| "loss": 0.5508, |
| "step": 640 |
| }, |
| { |
| "epoch": 1.954887218045113, |
| "grad_norm": 0.4783796037203571, |
| "learning_rate": 5e-06, |
| "loss": 0.5626, |
| "step": 650 |
| }, |
| { |
| "epoch": 1.9849624060150375, |
| "grad_norm": 0.4380429796439722, |
| "learning_rate": 5e-06, |
| "loss": 0.5579, |
| "step": 660 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_loss": 0.6012639403343201, |
| "eval_runtime": 116.0368, |
| "eval_samples_per_second": 77.2, |
| "eval_steps_per_second": 0.603, |
| "step": 665 |
| }, |
| { |
| "epoch": 2.0150375939849625, |
| "grad_norm": 0.8195061236201225, |
| "learning_rate": 5e-06, |
| "loss": 0.5349, |
| "step": 670 |
| }, |
| { |
| "epoch": 2.045112781954887, |
| "grad_norm": 0.5727072462132908, |
| "learning_rate": 5e-06, |
| "loss": 0.5023, |
| "step": 680 |
| }, |
| { |
| "epoch": 2.075187969924812, |
| "grad_norm": 0.5716937306059446, |
| "learning_rate": 5e-06, |
| "loss": 0.5025, |
| "step": 690 |
| }, |
| { |
| "epoch": 2.1052631578947367, |
| "grad_norm": 0.5936120169922254, |
| "learning_rate": 5e-06, |
| "loss": 0.502, |
| "step": 700 |
| }, |
| { |
| "epoch": 2.1353383458646618, |
| "grad_norm": 0.5323072505815232, |
| "learning_rate": 5e-06, |
| "loss": 0.5005, |
| "step": 710 |
| }, |
| { |
| "epoch": 2.1654135338345863, |
| "grad_norm": 0.5082968360924967, |
| "learning_rate": 5e-06, |
| "loss": 0.5037, |
| "step": 720 |
| }, |
| { |
| "epoch": 2.1954887218045114, |
| "grad_norm": 0.5227485772254855, |
| "learning_rate": 5e-06, |
| "loss": 0.4989, |
| "step": 730 |
| }, |
| { |
| "epoch": 2.225563909774436, |
| "grad_norm": 0.6533374147188858, |
| "learning_rate": 5e-06, |
| "loss": 0.5051, |
| "step": 740 |
| }, |
| { |
| "epoch": 2.255639097744361, |
| "grad_norm": 0.5912234992209703, |
| "learning_rate": 5e-06, |
| "loss": 0.5048, |
| "step": 750 |
| }, |
| { |
| "epoch": 2.2857142857142856, |
| "grad_norm": 0.5434227115064355, |
| "learning_rate": 5e-06, |
| "loss": 0.508, |
| "step": 760 |
| }, |
| { |
| "epoch": 2.3157894736842106, |
| "grad_norm": 0.5720087264790767, |
| "learning_rate": 5e-06, |
| "loss": 0.5099, |
| "step": 770 |
| }, |
| { |
| "epoch": 2.345864661654135, |
| "grad_norm": 0.6089476678610672, |
| "learning_rate": 5e-06, |
| "loss": 0.5024, |
| "step": 780 |
| }, |
| { |
| "epoch": 2.3759398496240602, |
| "grad_norm": 0.5508640915356684, |
| "learning_rate": 5e-06, |
| "loss": 0.5115, |
| "step": 790 |
| }, |
| { |
| "epoch": 2.406015037593985, |
| "grad_norm": 0.5118656252283122, |
| "learning_rate": 5e-06, |
| "loss": 0.5051, |
| "step": 800 |
| }, |
| { |
| "epoch": 2.43609022556391, |
| "grad_norm": 0.5065404177593222, |
| "learning_rate": 5e-06, |
| "loss": 0.5094, |
| "step": 810 |
| }, |
| { |
| "epoch": 2.4661654135338344, |
| "grad_norm": 0.5181429354716248, |
| "learning_rate": 5e-06, |
| "loss": 0.5106, |
| "step": 820 |
| }, |
| { |
| "epoch": 2.4962406015037595, |
| "grad_norm": 0.5322514457543749, |
| "learning_rate": 5e-06, |
| "loss": 0.5109, |
| "step": 830 |
| }, |
| { |
| "epoch": 2.526315789473684, |
| "grad_norm": 0.6196654734671412, |
| "learning_rate": 5e-06, |
| "loss": 0.5061, |
| "step": 840 |
| }, |
| { |
| "epoch": 2.556390977443609, |
| "grad_norm": 0.5256550356188423, |
| "learning_rate": 5e-06, |
| "loss": 0.5096, |
| "step": 850 |
| }, |
| { |
| "epoch": 2.5864661654135337, |
| "grad_norm": 0.4721653082846406, |
| "learning_rate": 5e-06, |
| "loss": 0.5091, |
| "step": 860 |
| }, |
| { |
| "epoch": 2.6165413533834587, |
| "grad_norm": 0.5461766886120224, |
| "learning_rate": 5e-06, |
| "loss": 0.5125, |
| "step": 870 |
| }, |
| { |
| "epoch": 2.6466165413533833, |
| "grad_norm": 0.8881592761313983, |
| "learning_rate": 5e-06, |
| "loss": 0.5133, |
| "step": 880 |
| }, |
| { |
| "epoch": 2.6766917293233083, |
| "grad_norm": 0.5215263169318092, |
| "learning_rate": 5e-06, |
| "loss": 0.5137, |
| "step": 890 |
| }, |
| { |
| "epoch": 2.706766917293233, |
| "grad_norm": 0.5970816519010325, |
| "learning_rate": 5e-06, |
| "loss": 0.5085, |
| "step": 900 |
| }, |
| { |
| "epoch": 2.736842105263158, |
| "grad_norm": 0.5663757119297533, |
| "learning_rate": 5e-06, |
| "loss": 0.5136, |
| "step": 910 |
| }, |
| { |
| "epoch": 2.7669172932330826, |
| "grad_norm": 0.48531820284070437, |
| "learning_rate": 5e-06, |
| "loss": 0.5105, |
| "step": 920 |
| }, |
| { |
| "epoch": 2.7969924812030076, |
| "grad_norm": 0.5235848494113168, |
| "learning_rate": 5e-06, |
| "loss": 0.5106, |
| "step": 930 |
| }, |
| { |
| "epoch": 2.827067669172932, |
| "grad_norm": 0.6230274670862681, |
| "learning_rate": 5e-06, |
| "loss": 0.5109, |
| "step": 940 |
| }, |
| { |
| "epoch": 2.857142857142857, |
| "grad_norm": 0.5464639670084257, |
| "learning_rate": 5e-06, |
| "loss": 0.502, |
| "step": 950 |
| }, |
| { |
| "epoch": 2.887218045112782, |
| "grad_norm": 0.5388668922788399, |
| "learning_rate": 5e-06, |
| "loss": 0.5059, |
| "step": 960 |
| }, |
| { |
| "epoch": 2.917293233082707, |
| "grad_norm": 0.5634658215547509, |
| "learning_rate": 5e-06, |
| "loss": 0.5147, |
| "step": 970 |
| }, |
| { |
| "epoch": 2.9473684210526314, |
| "grad_norm": 0.5876554762818471, |
| "learning_rate": 5e-06, |
| "loss": 0.507, |
| "step": 980 |
| }, |
| { |
| "epoch": 2.9774436090225564, |
| "grad_norm": 0.5816222818084577, |
| "learning_rate": 5e-06, |
| "loss": 0.5116, |
| "step": 990 |
| }, |
| { |
| "epoch": 2.995488721804511, |
| "eval_loss": 0.605595588684082, |
| "eval_runtime": 114.1196, |
| "eval_samples_per_second": 78.497, |
| "eval_steps_per_second": 0.613, |
| "step": 996 |
| }, |
| { |
| "epoch": 2.995488721804511, |
| "step": 996, |
| "total_flos": 1667918337146880.0, |
| "train_loss": 0.5664463261045126, |
| "train_runtime": 17372.9601, |
| "train_samples_per_second": 29.39, |
| "train_steps_per_second": 0.057 |
| } |
| ], |
| "logging_steps": 10, |
| "max_steps": 996, |
| "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": 1667918337146880.0, |
| "train_batch_size": 8, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
