{ "best_global_step": null, "best_metric": null, "best_model_checkpoint": null, "epoch": 0.8901746967842439, "eval_steps": 100, "global_step": 500, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "entropy": 1.3253214355558156, "epoch": 0.017803493935684877, "grad_norm": 14.5, "learning_rate": 6.000000000000001e-07, "loss": 2.1959, "mean_token_accuracy": 0.5975642748177051, "num_tokens": 259048.0, "step": 10 }, { "entropy": 1.3036176152527332, "epoch": 0.035606987871369754, "grad_norm": 12.75, "learning_rate": 1.2666666666666669e-06, "loss": 2.1499, "mean_token_accuracy": 0.6023047739639878, "num_tokens": 532019.0, "step": 20 }, { "entropy": 1.3484113823622466, "epoch": 0.053410481807054634, "grad_norm": 10.8125, "learning_rate": 1.9333333333333336e-06, "loss": 2.108, "mean_token_accuracy": 0.6078463492915034, "num_tokens": 807220.0, "step": 30 }, { "entropy": 1.4311681490391492, "epoch": 0.07121397574273951, "grad_norm": 7.875, "learning_rate": 2.6e-06, "loss": 2.0349, "mean_token_accuracy": 0.6107733339071274, "num_tokens": 1068429.0, "step": 40 }, { "entropy": 1.479667642340064, "epoch": 0.08901746967842439, "grad_norm": 6.4375, "learning_rate": 3.266666666666667e-06, "loss": 1.8649, "mean_token_accuracy": 0.630409375578165, "num_tokens": 1336698.0, "step": 50 }, { "entropy": 1.5030938494950532, "epoch": 0.10682096361410927, "grad_norm": 4.1875, "learning_rate": 3.9333333333333335e-06, "loss": 1.7217, "mean_token_accuracy": 0.6490663634613156, "num_tokens": 1608574.0, "step": 60 }, { "entropy": 1.5066865295171739, "epoch": 0.12462445754979415, "grad_norm": 2.71875, "learning_rate": 4.600000000000001e-06, "loss": 1.5716, "mean_token_accuracy": 0.6711439102888107, "num_tokens": 1872747.0, "step": 70 }, { "entropy": 1.3747932206839324, "epoch": 0.14242795148547902, "grad_norm": 2.234375, "learning_rate": 5.2666666666666665e-06, "loss": 1.4403, "mean_token_accuracy": 0.6936415441334247, "num_tokens": 2133173.0, "step": 80 }, { "entropy": 1.1736511185765266, "epoch": 0.1602314454211639, "grad_norm": 1.921875, "learning_rate": 5.933333333333335e-06, "loss": 1.2902, "mean_token_accuracy": 0.7222886722534895, "num_tokens": 2406145.0, "step": 90 }, { "entropy": 1.1317794900387526, "epoch": 0.17803493935684878, "grad_norm": 1.8046875, "learning_rate": 6.600000000000001e-06, "loss": 1.2424, "mean_token_accuracy": 0.7295983772724867, "num_tokens": 2663709.0, "step": 100 }, { "epoch": 0.17803493935684878, "eval_biology_entropy": 2.3133783559799195, "eval_biology_loss": 2.7910587787628174, "eval_biology_mean_token_accuracy": 0.5260232911109924, "eval_biology_num_tokens": 2663709.0, "eval_biology_runtime": 39.1656, "eval_biology_samples_per_second": 12.766, "eval_biology_steps_per_second": 3.192, "step": 100 }, { "epoch": 0.17803493935684878, "eval_chemistry_entropy": 1.0788086318969727, "eval_chemistry_loss": 1.2403336763381958, "eval_chemistry_mean_token_accuracy": 0.7357268543243408, "eval_chemistry_num_tokens": 2663709.0, "eval_chemistry_runtime": 48.7822, "eval_chemistry_samples_per_second": 10.25, "eval_chemistry_steps_per_second": 2.562, "step": 100 }, { "epoch": 0.17803493935684878, "eval_math_entropy": 0.938195451259613, "eval_math_loss": 1.3248047828674316, "eval_math_mean_token_accuracy": 0.7312402629852295, "eval_math_num_tokens": 2663709.0, "eval_math_runtime": 50.197, "eval_math_samples_per_second": 9.961, "eval_math_steps_per_second": 2.49, "step": 100 }, { "epoch": 0.17803493935684878, "eval_physics_entropy": 1.022246531009674, "eval_physics_loss": 1.176153302192688, "eval_physics_mean_token_accuracy": 0.7475795021057129, "eval_physics_num_tokens": 2663709.0, "eval_physics_runtime": 57.6895, "eval_physics_samples_per_second": 8.667, "eval_physics_steps_per_second": 2.167, "step": 100 }, { "entropy": 1.0380497876554728, "epoch": 0.19583843329253367, "grad_norm": 1.7421875, "learning_rate": 7.266666666666668e-06, "loss": 1.157, "mean_token_accuracy": 0.7445634145289659, "num_tokens": 2934980.0, "step": 110 }, { "entropy": 1.0149445166811346, "epoch": 0.21364192722821854, "grad_norm": 1.6015625, "learning_rate": 7.933333333333334e-06, "loss": 1.1167, "mean_token_accuracy": 0.7485828153789044, "num_tokens": 3208702.0, "step": 120 }, { "entropy": 0.9639404848217964, "epoch": 0.2314454211639034, "grad_norm": 1.6953125, "learning_rate": 8.6e-06, "loss": 1.0979, "mean_token_accuracy": 0.7553199239075183, "num_tokens": 3480119.0, "step": 130 }, { "entropy": 0.9687324531376362, "epoch": 0.2492489150995883, "grad_norm": 1.5625, "learning_rate": 9.266666666666667e-06, "loss": 1.0762, "mean_token_accuracy": 0.757096454873681, "num_tokens": 3746392.0, "step": 140 }, { "entropy": 0.9682734729722142, "epoch": 0.26705240903527316, "grad_norm": 1.875, "learning_rate": 9.933333333333334e-06, "loss": 1.0798, "mean_token_accuracy": 0.7543047085404396, "num_tokens": 4019487.0, "step": 150 }, { "entropy": 0.9270784268155694, "epoch": 0.28485590297095803, "grad_norm": 1.6015625, "learning_rate": 1.0600000000000002e-05, "loss": 1.047, "mean_token_accuracy": 0.7619738396257162, "num_tokens": 4298100.0, "step": 160 }, { "entropy": 0.9063327137380839, "epoch": 0.30265939690664295, "grad_norm": 1.5546875, "learning_rate": 1.1266666666666668e-05, "loss": 1.0199, "mean_token_accuracy": 0.7664302848279476, "num_tokens": 4568116.0, "step": 170 }, { "entropy": 0.9037324048578739, "epoch": 0.3204628908423278, "grad_norm": 1.7109375, "learning_rate": 1.1933333333333335e-05, "loss": 1.0144, "mean_token_accuracy": 0.7657378144562245, "num_tokens": 4831334.0, "step": 180 }, { "entropy": 0.8886868219822646, "epoch": 0.3382663847780127, "grad_norm": 1.3359375, "learning_rate": 1.2600000000000001e-05, "loss": 0.9874, "mean_token_accuracy": 0.7726409930735827, "num_tokens": 5103301.0, "step": 190 }, { "entropy": 0.8936551520600915, "epoch": 0.35606987871369755, "grad_norm": 1.3984375, "learning_rate": 1.3266666666666668e-05, "loss": 1.019, "mean_token_accuracy": 0.7648640714585782, "num_tokens": 5362915.0, "step": 200 }, { "epoch": 0.35606987871369755, "eval_biology_entropy": 2.0171296310424807, "eval_biology_loss": 2.3780677318573, "eval_biology_mean_token_accuracy": 0.5706260974407196, "eval_biology_num_tokens": 5362915.0, "eval_biology_runtime": 39.1203, "eval_biology_samples_per_second": 12.781, "eval_biology_steps_per_second": 3.195, "step": 200 }, { "epoch": 0.35606987871369755, "eval_chemistry_entropy": 0.8995157589912415, "eval_chemistry_loss": 1.022930383682251, "eval_chemistry_mean_token_accuracy": 0.7701377034187317, "eval_chemistry_num_tokens": 5362915.0, "eval_chemistry_runtime": 48.7263, "eval_chemistry_samples_per_second": 10.261, "eval_chemistry_steps_per_second": 2.565, "step": 200 }, { "epoch": 0.35606987871369755, "eval_math_entropy": 0.8352043969631195, "eval_math_loss": 1.1696399450302124, "eval_math_mean_token_accuracy": 0.7531440143585205, "eval_math_num_tokens": 5362915.0, "eval_math_runtime": 50.1494, "eval_math_samples_per_second": 9.97, "eval_math_steps_per_second": 2.493, "step": 200 }, { "epoch": 0.35606987871369755, "eval_physics_entropy": 0.8889413847923279, "eval_physics_loss": 0.9962027072906494, "eval_physics_mean_token_accuracy": 0.775176504611969, "eval_physics_num_tokens": 5362915.0, "eval_physics_runtime": 57.6533, "eval_physics_samples_per_second": 8.673, "eval_physics_steps_per_second": 2.168, "step": 200 }, { "entropy": 0.8509893301874399, "epoch": 0.3738733726493824, "grad_norm": 1.3359375, "learning_rate": 1.3933333333333334e-05, "loss": 0.9619, "mean_token_accuracy": 0.7772600032389164, "num_tokens": 5625455.0, "step": 210 }, { "entropy": 0.8560685662552714, "epoch": 0.39167686658506734, "grad_norm": 1.4609375, "learning_rate": 1.46e-05, "loss": 0.9666, "mean_token_accuracy": 0.7764565829187632, "num_tokens": 5890806.0, "step": 220 }, { "entropy": 0.8282079607248306, "epoch": 0.4094803605207522, "grad_norm": 1.3828125, "learning_rate": 1.5266666666666667e-05, "loss": 0.9353, "mean_token_accuracy": 0.7817027729004622, "num_tokens": 6158751.0, "step": 230 }, { "entropy": 0.8447791969403624, "epoch": 0.4272838544564371, "grad_norm": 1.4296875, "learning_rate": 1.5933333333333336e-05, "loss": 0.9428, "mean_token_accuracy": 0.7795210376381874, "num_tokens": 6423747.0, "step": 240 }, { "entropy": 0.8478817906230688, "epoch": 0.44508734839212194, "grad_norm": 1.8984375, "learning_rate": 1.66e-05, "loss": 0.9542, "mean_token_accuracy": 0.7772787544876337, "num_tokens": 6684517.0, "step": 250 }, { "entropy": 0.8135418241843582, "epoch": 0.4628908423278068, "grad_norm": 1.453125, "learning_rate": 1.726666666666667e-05, "loss": 0.9297, "mean_token_accuracy": 0.783412617444992, "num_tokens": 6945659.0, "step": 260 }, { "entropy": 0.8251218661665917, "epoch": 0.48069433626349173, "grad_norm": 1.6640625, "learning_rate": 1.7933333333333333e-05, "loss": 0.9307, "mean_token_accuracy": 0.7831020273268223, "num_tokens": 7217797.0, "step": 270 }, { "entropy": 0.7879339458420873, "epoch": 0.4984978301991766, "grad_norm": 1.328125, "learning_rate": 1.86e-05, "loss": 0.8795, "mean_token_accuracy": 0.7897050529718399, "num_tokens": 7498182.0, "step": 280 }, { "entropy": 0.8145477209240198, "epoch": 0.5163013241348615, "grad_norm": 1.265625, "learning_rate": 1.926666666666667e-05, "loss": 0.9187, "mean_token_accuracy": 0.7838029216974973, "num_tokens": 7761686.0, "step": 290 }, { "entropy": 0.8231891892850399, "epoch": 0.5341048180705463, "grad_norm": 1.5625, "learning_rate": 1.9933333333333334e-05, "loss": 0.9271, "mean_token_accuracy": 0.7819279827177524, "num_tokens": 8026923.0, "step": 300 }, { "epoch": 0.5341048180705463, "eval_biology_entropy": 1.8821040544509888, "eval_biology_loss": 2.226550340652466, "eval_biology_mean_token_accuracy": 0.5873979187011719, "eval_biology_num_tokens": 8026923.0, "eval_biology_runtime": 39.1096, "eval_biology_samples_per_second": 12.785, "eval_biology_steps_per_second": 3.196, "step": 300 }, { "epoch": 0.5341048180705463, "eval_chemistry_entropy": 0.7922393922805786, "eval_chemistry_loss": 0.9438055157661438, "eval_chemistry_mean_token_accuracy": 0.7841580080986023, "eval_chemistry_num_tokens": 8026923.0, "eval_chemistry_runtime": 48.7268, "eval_chemistry_samples_per_second": 10.261, "eval_chemistry_steps_per_second": 2.565, "step": 300 }, { "epoch": 0.5341048180705463, "eval_math_entropy": 0.7649447858333588, "eval_math_loss": 1.1145466566085815, "eval_math_mean_token_accuracy": 0.763204463005066, "eval_math_num_tokens": 8026923.0, "eval_math_runtime": 50.1205, "eval_math_samples_per_second": 9.976, "eval_math_steps_per_second": 2.494, "step": 300 }, { "epoch": 0.5341048180705463, "eval_physics_entropy": 0.7964776389598847, "eval_physics_loss": 0.9373180866241455, "eval_physics_mean_token_accuracy": 0.784841582775116, "eval_physics_num_tokens": 8026923.0, "eval_physics_runtime": 57.6222, "eval_physics_samples_per_second": 8.677, "eval_physics_steps_per_second": 2.169, "step": 300 }, { "entropy": 0.800599187426269, "epoch": 0.5519083120062312, "grad_norm": 1.6640625, "learning_rate": 1.9933333333333334e-05, "loss": 0.9071, "mean_token_accuracy": 0.7856069568544626, "num_tokens": 8301506.0, "step": 310 }, { "entropy": 0.7965721761807799, "epoch": 0.5697118059419161, "grad_norm": 1.234375, "learning_rate": 1.985925925925926e-05, "loss": 0.9139, "mean_token_accuracy": 0.786777526140213, "num_tokens": 8559126.0, "step": 320 }, { "entropy": 0.7925007566809654, "epoch": 0.587515299877601, "grad_norm": 1.4453125, "learning_rate": 1.9785185185185187e-05, "loss": 0.906, "mean_token_accuracy": 0.7864235289394855, "num_tokens": 8821088.0, "step": 330 }, { "entropy": 0.7728251311928034, "epoch": 0.6053187938132859, "grad_norm": 1.2109375, "learning_rate": 1.971111111111111e-05, "loss": 0.8764, "mean_token_accuracy": 0.7919386122375727, "num_tokens": 9089866.0, "step": 340 }, { "entropy": 0.7705478269606829, "epoch": 0.6231222877489707, "grad_norm": 1.3515625, "learning_rate": 1.963703703703704e-05, "loss": 0.8837, "mean_token_accuracy": 0.7896027479320764, "num_tokens": 9358866.0, "step": 350 }, { "entropy": 0.7656249281018972, "epoch": 0.6409257816846556, "grad_norm": 1.1875, "learning_rate": 1.9562962962962964e-05, "loss": 0.8626, "mean_token_accuracy": 0.7931958142668009, "num_tokens": 9629977.0, "step": 360 }, { "entropy": 0.7598591955378652, "epoch": 0.6587292756203404, "grad_norm": 1.296875, "learning_rate": 1.948888888888889e-05, "loss": 0.8608, "mean_token_accuracy": 0.7945109561085701, "num_tokens": 9897757.0, "step": 370 }, { "entropy": 0.7398761732503771, "epoch": 0.6765327695560254, "grad_norm": 1.265625, "learning_rate": 1.9414814814814817e-05, "loss": 0.848, "mean_token_accuracy": 0.7985429972410202, "num_tokens": 10168367.0, "step": 380 }, { "entropy": 0.7331193052232265, "epoch": 0.6943362634917103, "grad_norm": 1.4296875, "learning_rate": 1.9340740740740743e-05, "loss": 0.8366, "mean_token_accuracy": 0.8001827634871006, "num_tokens": 10439039.0, "step": 390 }, { "entropy": 0.7496613372117281, "epoch": 0.7121397574273951, "grad_norm": 1.15625, "learning_rate": 1.926666666666667e-05, "loss": 0.8517, "mean_token_accuracy": 0.7967161379754544, "num_tokens": 10709549.0, "step": 400 }, { "epoch": 0.7121397574273951, "eval_biology_entropy": 1.823248610496521, "eval_biology_loss": 2.159681558609009, "eval_biology_mean_token_accuracy": 0.5972214796543122, "eval_biology_num_tokens": 10709549.0, "eval_biology_runtime": 39.2024, "eval_biology_samples_per_second": 12.754, "eval_biology_steps_per_second": 3.189, "step": 400 }, { "epoch": 0.7121397574273951, "eval_chemistry_entropy": 0.7672218294143677, "eval_chemistry_loss": 0.8971230387687683, "eval_chemistry_mean_token_accuracy": 0.7928389143943787, "eval_chemistry_num_tokens": 10709549.0, "eval_chemistry_runtime": 48.8296, "eval_chemistry_samples_per_second": 10.24, "eval_chemistry_steps_per_second": 2.56, "step": 400 }, { "epoch": 0.7121397574273951, "eval_math_entropy": 0.7494234163761139, "eval_math_loss": 1.0849554538726807, "eval_math_mean_token_accuracy": 0.7666959943771362, "eval_math_num_tokens": 10709549.0, "eval_math_runtime": 50.2325, "eval_math_samples_per_second": 9.954, "eval_math_steps_per_second": 2.488, "step": 400 }, { "epoch": 0.7121397574273951, "eval_physics_entropy": 0.7731240901947022, "eval_physics_loss": 0.9018430709838867, "eval_physics_mean_token_accuracy": 0.7909709796905517, "eval_physics_num_tokens": 10709549.0, "eval_physics_runtime": 57.7809, "eval_physics_samples_per_second": 8.653, "eval_physics_steps_per_second": 2.163, "step": 400 }, { "entropy": 0.7354462265968322, "epoch": 0.72994325136308, "grad_norm": 1.3125, "learning_rate": 1.9192592592592593e-05, "loss": 0.8411, "mean_token_accuracy": 0.8003354988992214, "num_tokens": 10976011.0, "step": 410 }, { "entropy": 0.7435122076421976, "epoch": 0.7477467452987648, "grad_norm": 1.15625, "learning_rate": 1.911851851851852e-05, "loss": 0.8438, "mean_token_accuracy": 0.7971596572548151, "num_tokens": 11245251.0, "step": 420 }, { "entropy": 0.759309609234333, "epoch": 0.7655502392344498, "grad_norm": 1.375, "learning_rate": 1.9044444444444446e-05, "loss": 0.8579, "mean_token_accuracy": 0.792990094423294, "num_tokens": 11518514.0, "step": 430 }, { "entropy": 0.7545185785740613, "epoch": 0.7833537331701347, "grad_norm": 1.3125, "learning_rate": 1.8970370370370372e-05, "loss": 0.8574, "mean_token_accuracy": 0.794682926684618, "num_tokens": 11787117.0, "step": 440 }, { "entropy": 0.751254404336214, "epoch": 0.8011572271058195, "grad_norm": 1.515625, "learning_rate": 1.8896296296296295e-05, "loss": 0.8495, "mean_token_accuracy": 0.7972703959792853, "num_tokens": 12050993.0, "step": 450 }, { "entropy": 0.7543802969157696, "epoch": 0.8189607210415044, "grad_norm": 1.28125, "learning_rate": 1.8822222222222225e-05, "loss": 0.8527, "mean_token_accuracy": 0.7960003320127725, "num_tokens": 12311688.0, "step": 460 }, { "entropy": 0.7186065271496773, "epoch": 0.8367642149771892, "grad_norm": 1.15625, "learning_rate": 1.874814814814815e-05, "loss": 0.8096, "mean_token_accuracy": 0.8027401898056269, "num_tokens": 12598408.0, "step": 470 }, { "entropy": 0.7257912319153548, "epoch": 0.8545677089128741, "grad_norm": 1.125, "learning_rate": 1.8674074074074075e-05, "loss": 0.8299, "mean_token_accuracy": 0.8012042317539454, "num_tokens": 12869642.0, "step": 480 }, { "entropy": 0.7173270899802446, "epoch": 0.8723712028485591, "grad_norm": 1.125, "learning_rate": 1.86e-05, "loss": 0.8146, "mean_token_accuracy": 0.8032398864626884, "num_tokens": 13139555.0, "step": 490 }, { "entropy": 0.7487573977559805, "epoch": 0.8901746967842439, "grad_norm": 1.2421875, "learning_rate": 1.8525925925925928e-05, "loss": 0.8562, "mean_token_accuracy": 0.7955521304160357, "num_tokens": 13396545.0, "step": 500 }, { "epoch": 0.8901746967842439, "eval_biology_entropy": 1.8204833917617798, "eval_biology_loss": 2.1014022827148438, "eval_biology_mean_token_accuracy": 0.6051526765823364, "eval_biology_num_tokens": 13396545.0, "eval_biology_runtime": 39.1543, "eval_biology_samples_per_second": 12.77, "eval_biology_steps_per_second": 3.192, "step": 500 }, { "epoch": 0.8901746967842439, "eval_chemistry_entropy": 0.7538458592891694, "eval_chemistry_loss": 0.8699584007263184, "eval_chemistry_mean_token_accuracy": 0.7975720787048339, "eval_chemistry_num_tokens": 13396545.0, "eval_chemistry_runtime": 48.8229, "eval_chemistry_samples_per_second": 10.241, "eval_chemistry_steps_per_second": 2.56, "step": 500 }, { "epoch": 0.8901746967842439, "eval_math_entropy": 0.7398368701934814, "eval_math_loss": 1.0758970975875854, "eval_math_mean_token_accuracy": 0.7687290472984314, "eval_math_num_tokens": 13396545.0, "eval_math_runtime": 50.0916, "eval_math_samples_per_second": 9.982, "eval_math_steps_per_second": 2.495, "step": 500 }, { "epoch": 0.8901746967842439, "eval_physics_entropy": 0.7607374501228332, "eval_physics_loss": 0.8848790526390076, "eval_physics_mean_token_accuracy": 0.7937801394462586, "eval_physics_num_tokens": 13396545.0, "eval_physics_runtime": 57.6599, "eval_physics_samples_per_second": 8.672, "eval_physics_steps_per_second": 2.168, "step": 500 } ], "logging_steps": 10, "max_steps": 3000, "num_input_tokens_seen": 0, "num_train_epochs": 6, "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": 1.0399995579963064e+18, "train_batch_size": 4, "trial_name": null, "trial_params": null }