| { |
| "best_global_step": null, |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 5.0, |
| "eval_steps": 500, |
| "global_step": 2075, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.060350030175015085, |
| "grad_norm": 0.22048431634902954, |
| "learning_rate": 8.18181818181818e-05, |
| "loss": 1.8437, |
| "mean_token_accuracy": 0.6185667091608047, |
| "num_tokens": 157017.0, |
| "step": 25 |
| }, |
| { |
| "epoch": 0.12070006035003017, |
| "grad_norm": 0.28467562794685364, |
| "learning_rate": 0.00016704545454545452, |
| "loss": 1.0769, |
| "mean_token_accuracy": 0.7381608086824417, |
| "num_tokens": 282861.0, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.18105009052504525, |
| "grad_norm": 0.27485042810440063, |
| "learning_rate": 0.0002522727272727273, |
| "loss": 0.6762, |
| "mean_token_accuracy": 0.8168852615356446, |
| "num_tokens": 439048.0, |
| "step": 75 |
| }, |
| { |
| "epoch": 0.24140012070006034, |
| "grad_norm": 0.3554552495479584, |
| "learning_rate": 0.0002999887132933212, |
| "loss": 0.5642, |
| "mean_token_accuracy": 0.8437625843286515, |
| "num_tokens": 564700.0, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.30175015087507545, |
| "grad_norm": 0.24375054240226746, |
| "learning_rate": 0.0002998791256978121, |
| "loss": 0.4572, |
| "mean_token_accuracy": 0.869450313448906, |
| "num_tokens": 720573.0, |
| "step": 125 |
| }, |
| { |
| "epoch": 0.3621001810500905, |
| "grad_norm": 0.33107709884643555, |
| "learning_rate": 0.0002996530399366737, |
| "loss": 0.3565, |
| "mean_token_accuracy": 0.8982734948396682, |
| "num_tokens": 847195.0, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.4224502112251056, |
| "grad_norm": 0.30305179953575134, |
| "learning_rate": 0.00029931063174202567, |
| "loss": 0.2814, |
| "mean_token_accuracy": 0.9188993269205094, |
| "num_tokens": 1005729.0, |
| "step": 175 |
| }, |
| { |
| "epoch": 0.4828002414001207, |
| "grad_norm": 0.35704073309898376, |
| "learning_rate": 0.00029885216726118104, |
| "loss": 0.2394, |
| "mean_token_accuracy": 0.9293693268299102, |
| "num_tokens": 1132299.0, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.5431502715751357, |
| "grad_norm": 0.26589134335517883, |
| "learning_rate": 0.00029827800284977474, |
| "loss": 0.1821, |
| "mean_token_accuracy": 0.9475246322154999, |
| "num_tokens": 1291425.0, |
| "step": 225 |
| }, |
| { |
| "epoch": 0.6035003017501509, |
| "grad_norm": 0.3292100131511688, |
| "learning_rate": 0.00029758858479477575, |
| "loss": 0.1871, |
| "mean_token_accuracy": 0.9475667542219162, |
| "num_tokens": 1417320.0, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.663850331925166, |
| "grad_norm": 0.30751854181289673, |
| "learning_rate": 0.0002967844489675963, |
| "loss": 0.1468, |
| "mean_token_accuracy": 0.9586230200529099, |
| "num_tokens": 1575826.0, |
| "step": 275 |
| }, |
| { |
| "epoch": 0.724200362100181, |
| "grad_norm": 0.28926485776901245, |
| "learning_rate": 0.00029586622040756957, |
| "loss": 0.1372, |
| "mean_token_accuracy": 0.9609356409311295, |
| "num_tokens": 1702190.0, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.7845503922751962, |
| "grad_norm": 0.2157151848077774, |
| "learning_rate": 0.0002948346128361186, |
| "loss": 0.1159, |
| "mean_token_accuracy": 0.9670810961723327, |
| "num_tokens": 1860388.0, |
| "step": 325 |
| }, |
| { |
| "epoch": 0.8449004224502112, |
| "grad_norm": 0.3129127621650696, |
| "learning_rate": 0.00029369042810199416, |
| "loss": 0.0997, |
| "mean_token_accuracy": 0.9725474363565445, |
| "num_tokens": 1988000.0, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.9052504526252263, |
| "grad_norm": 0.12445805966854095, |
| "learning_rate": 0.0002924345555580135, |
| "loss": 0.099, |
| "mean_token_accuracy": 0.9723707520961762, |
| "num_tokens": 2146759.0, |
| "step": 375 |
| }, |
| { |
| "epoch": 0.9656004828002414, |
| "grad_norm": 0.2350274920463562, |
| "learning_rate": 0.000291067971369783, |
| "loss": 0.087, |
| "mean_token_accuracy": 0.9752390831708908, |
| "num_tokens": 2274851.0, |
| "step": 400 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_loss": 0.08540041744709015, |
| "eval_mean_token_accuracy": 0.9763027584230578, |
| "eval_num_tokens": 2354180.0, |
| "eval_runtime": 62.666, |
| "eval_samples_per_second": 5.888, |
| "eval_steps_per_second": 2.952, |
| "step": 415 |
| }, |
| { |
| "epoch": 1.024140012070006, |
| "grad_norm": 0.1618579775094986, |
| "learning_rate": 0.0002895917377569438, |
| "loss": 0.0946, |
| "mean_token_accuracy": 0.9745458819202542, |
| "num_tokens": 2424550.0, |
| "step": 425 |
| }, |
| { |
| "epoch": 1.0844900422450212, |
| "grad_norm": 0.17064224183559418, |
| "learning_rate": 0.00028800700216752875, |
| "loss": 0.059, |
| "mean_token_accuracy": 0.9832033175230026, |
| "num_tokens": 2565889.0, |
| "step": 450 |
| }, |
| { |
| "epoch": 1.1448400724200363, |
| "grad_norm": 0.15575812757015228, |
| "learning_rate": 0.00028631499638607285, |
| "loss": 0.0719, |
| "mean_token_accuracy": 0.9796904844045639, |
| "num_tokens": 2707834.0, |
| "step": 475 |
| }, |
| { |
| "epoch": 1.2051901025950513, |
| "grad_norm": 0.15536518394947052, |
| "learning_rate": 0.0002845170355761712, |
| "loss": 0.0586, |
| "mean_token_accuracy": 0.9835732167959214, |
| "num_tokens": 2850768.0, |
| "step": 500 |
| }, |
| { |
| "epoch": 1.2655401327700664, |
| "grad_norm": 0.13807828724384308, |
| "learning_rate": 0.0002826145172582274, |
| "loss": 0.0645, |
| "mean_token_accuracy": 0.9819038778543472, |
| "num_tokens": 2990877.0, |
| "step": 525 |
| }, |
| { |
| "epoch": 1.3258901629450814, |
| "grad_norm": 0.2444400042295456, |
| "learning_rate": 0.00028060892022318764, |
| "loss": 0.049, |
| "mean_token_accuracy": 0.9859754049777985, |
| "num_tokens": 3131813.0, |
| "step": 550 |
| }, |
| { |
| "epoch": 1.3862401931200965, |
| "grad_norm": 0.16260553896427155, |
| "learning_rate": 0.0002785018033831051, |
| "loss": 0.0724, |
| "mean_token_accuracy": 0.9795020925998688, |
| "num_tokens": 3271944.0, |
| "step": 575 |
| }, |
| { |
| "epoch": 1.4465902232951118, |
| "grad_norm": 0.15038788318634033, |
| "learning_rate": 0.0002762948045594276, |
| "loss": 0.048, |
| "mean_token_accuracy": 0.986338136792183, |
| "num_tokens": 3415529.0, |
| "step": 600 |
| }, |
| { |
| "epoch": 1.5069402534701268, |
| "grad_norm": 0.12307009845972061, |
| "learning_rate": 0.0002739896392099502, |
| "loss": 0.0648, |
| "mean_token_accuracy": 0.9821051049232483, |
| "num_tokens": 3557697.0, |
| "step": 625 |
| }, |
| { |
| "epoch": 1.567290283645142, |
| "grad_norm": 0.17994529008865356, |
| "learning_rate": 0.00027158809909542307, |
| "loss": 0.0458, |
| "mean_token_accuracy": 0.9869296491146088, |
| "num_tokens": 3701711.0, |
| "step": 650 |
| }, |
| { |
| "epoch": 1.627640313820157, |
| "grad_norm": 0.12831450998783112, |
| "learning_rate": 0.00026909205088685, |
| "loss": 0.0602, |
| "mean_token_accuracy": 0.9828966480493545, |
| "num_tokens": 3844440.0, |
| "step": 675 |
| }, |
| { |
| "epoch": 1.687990343995172, |
| "grad_norm": 0.12778432667255402, |
| "learning_rate": 0.0002665034347145612, |
| "loss": 0.0492, |
| "mean_token_accuracy": 0.9861533933877945, |
| "num_tokens": 3986855.0, |
| "step": 700 |
| }, |
| { |
| "epoch": 1.748340374170187, |
| "grad_norm": 0.09499543905258179, |
| "learning_rate": 0.000263824262660187, |
| "loss": 0.0561, |
| "mean_token_accuracy": 0.984484378695488, |
| "num_tokens": 4128398.0, |
| "step": 725 |
| }, |
| { |
| "epoch": 1.8086904043452021, |
| "grad_norm": 0.15675826370716095, |
| "learning_rate": 0.0002610566171927056, |
| "loss": 0.0426, |
| "mean_token_accuracy": 0.9876346814632416, |
| "num_tokens": 4270406.0, |
| "step": 750 |
| }, |
| { |
| "epoch": 1.8690404345202172, |
| "grad_norm": 0.11688172817230225, |
| "learning_rate": 0.00025820264954977976, |
| "loss": 0.0565, |
| "mean_token_accuracy": 0.9839567536115646, |
| "num_tokens": 4412708.0, |
| "step": 775 |
| }, |
| { |
| "epoch": 1.9293904646952322, |
| "grad_norm": 0.1486789733171463, |
| "learning_rate": 0.00025526457806564136, |
| "loss": 0.0417, |
| "mean_token_accuracy": 0.9880886596441268, |
| "num_tokens": 4556150.0, |
| "step": 800 |
| }, |
| { |
| "epoch": 1.9897404948702473, |
| "grad_norm": 0.11592721939086914, |
| "learning_rate": 0.00025224468644682245, |
| "loss": 0.0492, |
| "mean_token_accuracy": 0.9864854019880295, |
| "num_tokens": 4688551.0, |
| "step": 825 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_loss": 0.054148148745298386, |
| "eval_mean_token_accuracy": 0.9855220913887024, |
| "eval_num_tokens": 4708360.0, |
| "eval_runtime": 62.7544, |
| "eval_samples_per_second": 5.88, |
| "eval_steps_per_second": 2.948, |
| "step": 830 |
| }, |
| { |
| "epoch": 2.048280024140012, |
| "grad_norm": 0.1103988066315651, |
| "learning_rate": 0.00024914532199707444, |
| "loss": 0.0456, |
| "mean_token_accuracy": 0.9862007431148254, |
| "num_tokens": 4838545.0, |
| "step": 850 |
| }, |
| { |
| "epoch": 2.1086300543150274, |
| "grad_norm": 0.17865443229675293, |
| "learning_rate": 0.00024596889379285353, |
| "loss": 0.0335, |
| "mean_token_accuracy": 0.9900023031234741, |
| "num_tokens": 4971291.0, |
| "step": 875 |
| }, |
| { |
| "epoch": 2.1689800844900424, |
| "grad_norm": 0.08458594232797623, |
| "learning_rate": 0.00024271787081079228, |
| "loss": 0.0445, |
| "mean_token_accuracy": 0.9871802502870559, |
| "num_tokens": 5122538.0, |
| "step": 900 |
| }, |
| { |
| "epoch": 2.2293301146650575, |
| "grad_norm": 0.0892493948340416, |
| "learning_rate": 0.00023939478000861117, |
| "loss": 0.0347, |
| "mean_token_accuracy": 0.9895845329761506, |
| "num_tokens": 5256508.0, |
| "step": 925 |
| }, |
| { |
| "epoch": 2.2896801448400725, |
| "grad_norm": 0.08608334511518478, |
| "learning_rate": 0.00023600220436096318, |
| "loss": 0.0428, |
| "mean_token_accuracy": 0.9870429944992065, |
| "num_tokens": 5406644.0, |
| "step": 950 |
| }, |
| { |
| "epoch": 2.3500301750150876, |
| "grad_norm": 0.11120013892650604, |
| "learning_rate": 0.00023254278085173684, |
| "loss": 0.0304, |
| "mean_token_accuracy": 0.990639505982399, |
| "num_tokens": 5539678.0, |
| "step": 975 |
| }, |
| { |
| "epoch": 2.4103802051901027, |
| "grad_norm": 0.08932222425937653, |
| "learning_rate": 0.00022901919842437972, |
| "loss": 0.0453, |
| "mean_token_accuracy": 0.9866666996479034, |
| "num_tokens": 5692025.0, |
| "step": 1000 |
| }, |
| { |
| "epoch": 2.4707302353651177, |
| "grad_norm": 0.07231656461954117, |
| "learning_rate": 0.00022543419589183397, |
| "loss": 0.031, |
| "mean_token_accuracy": 0.990651159286499, |
| "num_tokens": 5824644.0, |
| "step": 1025 |
| }, |
| { |
| "epoch": 2.5310802655401328, |
| "grad_norm": 0.09120254963636398, |
| "learning_rate": 0.00022179055980770993, |
| "loss": 0.0421, |
| "mean_token_accuracy": 0.9870923537015915, |
| "num_tokens": 5974858.0, |
| "step": 1050 |
| }, |
| { |
| "epoch": 2.591430295715148, |
| "grad_norm": 0.09927839785814285, |
| "learning_rate": 0.0002180911223003513, |
| "loss": 0.0294, |
| "mean_token_accuracy": 0.9910473513603211, |
| "num_tokens": 6106888.0, |
| "step": 1075 |
| }, |
| { |
| "epoch": 2.651780325890163, |
| "grad_norm": 0.1529274433851242, |
| "learning_rate": 0.00021433875887147627, |
| "loss": 0.0398, |
| "mean_token_accuracy": 0.9881814271211624, |
| "num_tokens": 6257087.0, |
| "step": 1100 |
| }, |
| { |
| "epoch": 2.712130356065178, |
| "grad_norm": 0.07802440971136093, |
| "learning_rate": 0.00021053638616110525, |
| "loss": 0.0313, |
| "mean_token_accuracy": 0.9907257658243179, |
| "num_tokens": 6390499.0, |
| "step": 1125 |
| }, |
| { |
| "epoch": 2.772480386240193, |
| "grad_norm": 0.06188732013106346, |
| "learning_rate": 0.00020668695968051274, |
| "loss": 0.0463, |
| "mean_token_accuracy": 0.9869219189882279, |
| "num_tokens": 6542392.0, |
| "step": 1150 |
| }, |
| { |
| "epoch": 2.832830416415208, |
| "grad_norm": 0.09649744629859924, |
| "learning_rate": 0.00020279347151496482, |
| "loss": 0.0278, |
| "mean_token_accuracy": 0.9914505350589752, |
| "num_tokens": 6675191.0, |
| "step": 1175 |
| }, |
| { |
| "epoch": 2.8931804465902236, |
| "grad_norm": 0.05005449801683426, |
| "learning_rate": 0.00019885894799802922, |
| "loss": 0.0452, |
| "mean_token_accuracy": 0.9864512634277344, |
| "num_tokens": 6828887.0, |
| "step": 1200 |
| }, |
| { |
| "epoch": 2.9535304767652386, |
| "grad_norm": 0.09346788376569748, |
| "learning_rate": 0.00019488644735926396, |
| "loss": 0.0317, |
| "mean_token_accuracy": 0.9907874202728272, |
| "num_tokens": 6962022.0, |
| "step": 1225 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_loss": 0.04395000636577606, |
| "eval_mean_token_accuracy": 0.9879707652169305, |
| "eval_num_tokens": 7062540.0, |
| "eval_runtime": 62.7009, |
| "eval_samples_per_second": 5.885, |
| "eval_steps_per_second": 2.951, |
| "step": 1245 |
| }, |
| { |
| "epoch": 3.012070006035003, |
| "grad_norm": 0.10707546025514603, |
| "learning_rate": 0.00019087905734711452, |
| "loss": 0.0388, |
| "mean_token_accuracy": 0.9883427564630803, |
| "num_tokens": 7099771.0, |
| "step": 1250 |
| }, |
| { |
| "epoch": 3.0724200362100182, |
| "grad_norm": 0.04371380805969238, |
| "learning_rate": 0.00018683989282886613, |
| "loss": 0.0259, |
| "mean_token_accuracy": 0.9917033529281616, |
| "num_tokens": 7247681.0, |
| "step": 1275 |
| }, |
| { |
| "epoch": 3.1327700663850333, |
| "grad_norm": 0.0800153911113739, |
| "learning_rate": 0.0001827720933695173, |
| "loss": 0.0308, |
| "mean_token_accuracy": 0.9905261534452439, |
| "num_tokens": 7381955.0, |
| "step": 1300 |
| }, |
| { |
| "epoch": 3.1931200965600484, |
| "grad_norm": 0.09413493424654007, |
| "learning_rate": 0.00017867882079145627, |
| "loss": 0.0263, |
| "mean_token_accuracy": 0.9920635151863099, |
| "num_tokens": 7528804.0, |
| "step": 1325 |
| }, |
| { |
| "epoch": 3.2534701267350634, |
| "grad_norm": 0.07196678966283798, |
| "learning_rate": 0.00017456325671683724, |
| "loss": 0.0319, |
| "mean_token_accuracy": 0.9904261428117752, |
| "num_tokens": 7664857.0, |
| "step": 1350 |
| }, |
| { |
| "epoch": 3.3138201569100785, |
| "grad_norm": 0.07312282174825668, |
| "learning_rate": 0.00017042860009456638, |
| "loss": 0.0267, |
| "mean_token_accuracy": 0.9914035642147064, |
| "num_tokens": 7812961.0, |
| "step": 1375 |
| }, |
| { |
| "epoch": 3.3741701870850935, |
| "grad_norm": 0.07159588485956192, |
| "learning_rate": 0.00016627806471382066, |
| "loss": 0.0282, |
| "mean_token_accuracy": 0.9908248996734619, |
| "num_tokens": 7948932.0, |
| "step": 1400 |
| }, |
| { |
| "epoch": 3.4345202172601086, |
| "grad_norm": 0.04767977446317673, |
| "learning_rate": 0.00016211487670603078, |
| "loss": 0.0259, |
| "mean_token_accuracy": 0.9916666203737259, |
| "num_tokens": 8097306.0, |
| "step": 1425 |
| }, |
| { |
| "epoch": 3.4948702474351236, |
| "grad_norm": 0.061906732618808746, |
| "learning_rate": 0.0001579422720372715, |
| "loss": 0.0269, |
| "mean_token_accuracy": 0.9917305141687394, |
| "num_tokens": 8232788.0, |
| "step": 1450 |
| }, |
| { |
| "epoch": 3.5552202776101387, |
| "grad_norm": 0.04637061804533005, |
| "learning_rate": 0.00015376349399300745, |
| "loss": 0.0263, |
| "mean_token_accuracy": 0.9916825985908508, |
| "num_tokens": 8381745.0, |
| "step": 1475 |
| }, |
| { |
| "epoch": 3.6155703077851538, |
| "grad_norm": 0.12033607810735703, |
| "learning_rate": 0.0001495817906571492, |
| "loss": 0.0322, |
| "mean_token_accuracy": 0.9897854518890381, |
| "num_tokens": 8519204.0, |
| "step": 1500 |
| }, |
| { |
| "epoch": 3.675920337960169, |
| "grad_norm": 0.06281058490276337, |
| "learning_rate": 0.00014540041238738055, |
| "loss": 0.0249, |
| "mean_token_accuracy": 0.9918615108728409, |
| "num_tokens": 8666692.0, |
| "step": 1525 |
| }, |
| { |
| "epoch": 3.736270368135184, |
| "grad_norm": 0.06974755227565765, |
| "learning_rate": 0.00014122260928871734, |
| "loss": 0.0299, |
| "mean_token_accuracy": 0.9906923991441726, |
| "num_tokens": 8802593.0, |
| "step": 1550 |
| }, |
| { |
| "epoch": 3.796620398310199, |
| "grad_norm": 0.048978209495544434, |
| "learning_rate": 0.00013705162868726396, |
| "loss": 0.0258, |
| "mean_token_accuracy": 0.9917961621284485, |
| "num_tokens": 8951230.0, |
| "step": 1575 |
| }, |
| { |
| "epoch": 3.856970428485214, |
| "grad_norm": 0.0680331289768219, |
| "learning_rate": 0.00013289071260612855, |
| "loss": 0.0269, |
| "mean_token_accuracy": 0.991482435464859, |
| "num_tokens": 9086196.0, |
| "step": 1600 |
| }, |
| { |
| "epoch": 3.9173204586602295, |
| "grad_norm": 0.04798055812716484, |
| "learning_rate": 0.00012874309524546083, |
| "loss": 0.025, |
| "mean_token_accuracy": 0.9918987530469895, |
| "num_tokens": 9235448.0, |
| "step": 1625 |
| }, |
| { |
| "epoch": 3.9776704888352445, |
| "grad_norm": 0.07034997642040253, |
| "learning_rate": 0.00012461200046857084, |
| "loss": 0.0244, |
| "mean_token_accuracy": 0.9922022837400436, |
| "num_tokens": 9368109.0, |
| "step": 1650 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_loss": 0.04039536416530609, |
| "eval_mean_token_accuracy": 0.9890036863249702, |
| "eval_num_tokens": 9416720.0, |
| "eval_runtime": 62.709, |
| "eval_samples_per_second": 5.884, |
| "eval_steps_per_second": 2.95, |
| "step": 1660 |
| }, |
| { |
| "epoch": 4.036210018105009, |
| "grad_norm": 0.04823560640215874, |
| "learning_rate": 0.00012050063929608123, |
| "loss": 0.0245, |
| "mean_token_accuracy": 0.9920303588060989, |
| "num_tokens": 9515827.0, |
| "step": 1675 |
| }, |
| { |
| "epoch": 4.096560048280024, |
| "grad_norm": 0.04038386419415474, |
| "learning_rate": 0.0001164122074100633, |
| "loss": 0.0205, |
| "mean_token_accuracy": 0.9934929460287094, |
| "num_tokens": 9653322.0, |
| "step": 1700 |
| }, |
| { |
| "epoch": 4.15691007845504, |
| "grad_norm": 0.07340297102928162, |
| "learning_rate": 0.00011234988267009415, |
| "loss": 0.024, |
| "mean_token_accuracy": 0.9922641849517823, |
| "num_tokens": 9799350.0, |
| "step": 1725 |
| }, |
| { |
| "epoch": 4.217260108630055, |
| "grad_norm": 0.038047417998313904, |
| "learning_rate": 0.00010831682264316787, |
| "loss": 0.0199, |
| "mean_token_accuracy": 0.9933889234066009, |
| "num_tokens": 9936097.0, |
| "step": 1750 |
| }, |
| { |
| "epoch": 4.27761013880507, |
| "grad_norm": 0.13129326701164246, |
| "learning_rate": 0.00010431616214937911, |
| "loss": 0.0233, |
| "mean_token_accuracy": 0.9926201468706131, |
| "num_tokens": 10082564.0, |
| "step": 1775 |
| }, |
| { |
| "epoch": 4.337960168980085, |
| "grad_norm": 0.037602320313453674, |
| "learning_rate": 0.00010035101082528777, |
| "loss": 0.0205, |
| "mean_token_accuracy": 0.9932494872808456, |
| "num_tokens": 10219755.0, |
| "step": 1800 |
| }, |
| { |
| "epoch": 4.3983101991551, |
| "grad_norm": 0.04699968174099922, |
| "learning_rate": 9.642445070685809e-05, |
| "loss": 0.0245, |
| "mean_token_accuracy": 0.992087431550026, |
| "num_tokens": 10367027.0, |
| "step": 1825 |
| }, |
| { |
| "epoch": 4.458660229330115, |
| "grad_norm": 0.04391399025917053, |
| "learning_rate": 9.253953383385157e-05, |
| "loss": 0.0196, |
| "mean_token_accuracy": 0.9935489571094513, |
| "num_tokens": 10506017.0, |
| "step": 1850 |
| }, |
| { |
| "epoch": 4.51901025950513, |
| "grad_norm": 0.04784134775400162, |
| "learning_rate": 8.869927987753459e-05, |
| "loss": 0.0232, |
| "mean_token_accuracy": 0.9924498957395553, |
| "num_tokens": 10652825.0, |
| "step": 1875 |
| }, |
| { |
| "epoch": 4.579360289680145, |
| "grad_norm": 0.04309968650341034, |
| "learning_rate": 8.490667379354661e-05, |
| "loss": 0.0189, |
| "mean_token_accuracy": 0.9937076556682587, |
| "num_tokens": 10790476.0, |
| "step": 1900 |
| }, |
| { |
| "epoch": 4.63971031985516, |
| "grad_norm": 0.07193579524755478, |
| "learning_rate": 8.116466350175079e-05, |
| "loss": 0.0235, |
| "mean_token_accuracy": 0.992224183678627, |
| "num_tokens": 10935736.0, |
| "step": 1925 |
| }, |
| { |
| "epoch": 4.700060350030175, |
| "grad_norm": 0.04449706897139549, |
| "learning_rate": 7.747615759487304e-05, |
| "loss": 0.0194, |
| "mean_token_accuracy": 0.9936429464817047, |
| "num_tokens": 11072468.0, |
| "step": 1950 |
| }, |
| { |
| "epoch": 4.76041038020519, |
| "grad_norm": 0.06068187579512596, |
| "learning_rate": 7.38440230777085e-05, |
| "loss": 0.0235, |
| "mean_token_accuracy": 0.992416363954544, |
| "num_tokens": 11218509.0, |
| "step": 1975 |
| }, |
| { |
| "epoch": 4.820760410380205, |
| "grad_norm": 0.030500777065753937, |
| "learning_rate": 7.027108313865378e-05, |
| "loss": 0.0185, |
| "mean_token_accuracy": 0.9937655937671661, |
| "num_tokens": 11356841.0, |
| "step": 2000 |
| }, |
| { |
| "epoch": 4.88111044055522, |
| "grad_norm": 0.1406712532043457, |
| "learning_rate": 6.676011495529687e-05, |
| "loss": 0.0243, |
| "mean_token_accuracy": 0.9923334455490113, |
| "num_tokens": 11505617.0, |
| "step": 2025 |
| }, |
| { |
| "epoch": 4.941460470730235, |
| "grad_norm": 0.0490015484392643, |
| "learning_rate": 6.331384753577056e-05, |
| "loss": 0.0188, |
| "mean_token_accuracy": 0.993909627199173, |
| "num_tokens": 11642628.0, |
| "step": 2050 |
| }, |
| { |
| "epoch": 5.0, |
| "grad_norm": 0.13373523950576782, |
| "learning_rate": 5.993495959754631e-05, |
| "loss": 0.0221, |
| "mean_token_accuracy": 0.992490774577426, |
| "num_tokens": 11770900.0, |
| "step": 2075 |
| }, |
| { |
| "epoch": 5.0, |
| "eval_loss": 0.03932187706232071, |
| "eval_mean_token_accuracy": 0.9896563478418299, |
| "eval_num_tokens": 11770900.0, |
| "eval_runtime": 62.7147, |
| "eval_samples_per_second": 5.884, |
| "eval_steps_per_second": 2.95, |
| "step": 2075 |
| } |
| ], |
| "logging_steps": 25, |
| "max_steps": 2905, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 7, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 5.0566765205027635e+17, |
| "train_batch_size": 2, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
