4d8dcm2s / checkpoint-500 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
76cba56 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.2,
"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.0975568080320954,
"epoch": 0.004,
"grad_norm": 1928.0,
"learning_rate": 4.5000000000000003e-07,
"loss": 51.036,
"mean_token_accuracy": 0.5464097319170833,
"num_tokens": 51831.0,
"step": 10
},
{
"entropy": 1.2103175908327102,
"epoch": 0.008,
"grad_norm": 1012.0,
"learning_rate": 9.500000000000001e-07,
"loss": 47.7535,
"mean_token_accuracy": 0.549801179766655,
"num_tokens": 103172.0,
"step": 20
},
{
"entropy": 1.4519254021346568,
"epoch": 0.012,
"grad_norm": 1024.0,
"learning_rate": 1.45e-06,
"loss": 45.5399,
"mean_token_accuracy": 0.5487965773791075,
"num_tokens": 147915.0,
"step": 30
},
{
"entropy": 1.515171455964446,
"epoch": 0.016,
"grad_norm": 440.0,
"learning_rate": 1.9500000000000004e-06,
"loss": 40.0298,
"mean_token_accuracy": 0.5707932172343135,
"num_tokens": 195524.0,
"step": 40
},
{
"entropy": 1.529548142105341,
"epoch": 0.02,
"grad_norm": 328.0,
"learning_rate": 2.4500000000000003e-06,
"loss": 33.3046,
"mean_token_accuracy": 0.5914752574637532,
"num_tokens": 243747.0,
"step": 50
},
{
"entropy": 1.5671692434698343,
"epoch": 0.024,
"grad_norm": 241.0,
"learning_rate": 2.95e-06,
"loss": 30.3037,
"mean_token_accuracy": 0.6058902809396386,
"num_tokens": 296999.0,
"step": 60
},
{
"entropy": 1.549248830974102,
"epoch": 0.028,
"grad_norm": 235.0,
"learning_rate": 3.45e-06,
"loss": 29.0295,
"mean_token_accuracy": 0.6220679895952343,
"num_tokens": 342114.0,
"step": 70
},
{
"entropy": 1.42005056142807,
"epoch": 0.032,
"grad_norm": 167.0,
"learning_rate": 3.95e-06,
"loss": 26.6089,
"mean_token_accuracy": 0.6380933992564678,
"num_tokens": 390903.0,
"step": 80
},
{
"entropy": 1.5127948723733424,
"epoch": 0.036,
"grad_norm": 167.0,
"learning_rate": 4.450000000000001e-06,
"loss": 24.8172,
"mean_token_accuracy": 0.6465719180181623,
"num_tokens": 440359.0,
"step": 90
},
{
"entropy": 1.5200586583465339,
"epoch": 0.04,
"grad_norm": 142.0,
"learning_rate": 4.95e-06,
"loss": 24.207,
"mean_token_accuracy": 0.6497391745448112,
"num_tokens": 488481.0,
"step": 100
},
{
"epoch": 0.04,
"eval_biology_entropy": 1.4843950763940812,
"eval_biology_loss": 1.4266694784164429,
"eval_biology_mean_token_accuracy": 0.6619762739539147,
"eval_biology_num_tokens": 488481.0,
"eval_biology_runtime": 36.5287,
"eval_biology_samples_per_second": 13.688,
"eval_biology_steps_per_second": 13.688,
"step": 100
},
{
"epoch": 0.04,
"eval_chemistry_entropy": 1.091414896786213,
"eval_chemistry_loss": 1.0560342073440552,
"eval_chemistry_mean_token_accuracy": 0.7318743930459023,
"eval_chemistry_num_tokens": 488481.0,
"eval_chemistry_runtime": 39.96,
"eval_chemistry_samples_per_second": 12.513,
"eval_chemistry_steps_per_second": 12.513,
"step": 100
},
{
"epoch": 0.04,
"eval_math_entropy": 0.9103146089911461,
"eval_math_loss": 1.0316376686096191,
"eval_math_mean_token_accuracy": 0.7397149602174758,
"eval_math_num_tokens": 488481.0,
"eval_math_runtime": 41.0822,
"eval_math_samples_per_second": 12.171,
"eval_math_steps_per_second": 12.171,
"step": 100
},
{
"epoch": 0.04,
"eval_physics_entropy": 0.9860039167404174,
"eval_physics_loss": 0.9796567559242249,
"eval_physics_mean_token_accuracy": 0.7483667035698891,
"eval_physics_num_tokens": 488481.0,
"eval_physics_runtime": 45.3249,
"eval_physics_samples_per_second": 11.031,
"eval_physics_steps_per_second": 11.031,
"step": 100
},
{
"entropy": 1.444656402617693,
"epoch": 0.044,
"grad_norm": 121.0,
"learning_rate": 5.450000000000001e-06,
"loss": 22.4665,
"mean_token_accuracy": 0.6580426702275872,
"num_tokens": 537471.0,
"step": 110
},
{
"entropy": 1.3339354760944844,
"epoch": 0.048,
"grad_norm": 129.0,
"learning_rate": 5.950000000000001e-06,
"loss": 21.4085,
"mean_token_accuracy": 0.6670540764927864,
"num_tokens": 589868.0,
"step": 120
},
{
"entropy": 1.3668562419712544,
"epoch": 0.052,
"grad_norm": 124.5,
"learning_rate": 6.450000000000001e-06,
"loss": 21.7438,
"mean_token_accuracy": 0.6594993283972144,
"num_tokens": 639512.0,
"step": 130
},
{
"entropy": 1.3065196067094802,
"epoch": 0.056,
"grad_norm": 118.5,
"learning_rate": 6.95e-06,
"loss": 20.9541,
"mean_token_accuracy": 0.6683343779295683,
"num_tokens": 690754.0,
"step": 140
},
{
"entropy": 1.3160824533551931,
"epoch": 0.06,
"grad_norm": 115.5,
"learning_rate": 7.450000000000001e-06,
"loss": 21.525,
"mean_token_accuracy": 0.664486956782639,
"num_tokens": 737075.0,
"step": 150
},
{
"entropy": 1.3052487019449472,
"epoch": 0.064,
"grad_norm": 113.0,
"learning_rate": 7.950000000000002e-06,
"loss": 20.5473,
"mean_token_accuracy": 0.672793828509748,
"num_tokens": 791190.0,
"step": 160
},
{
"entropy": 1.29992272593081,
"epoch": 0.068,
"grad_norm": 100.0,
"learning_rate": 8.45e-06,
"loss": 21.0557,
"mean_token_accuracy": 0.6683677529916168,
"num_tokens": 841344.0,
"step": 170
},
{
"entropy": 1.3232017546892165,
"epoch": 0.072,
"grad_norm": 114.5,
"learning_rate": 8.95e-06,
"loss": 21.0887,
"mean_token_accuracy": 0.6686861604452133,
"num_tokens": 890075.0,
"step": 180
},
{
"entropy": 1.2425255428999662,
"epoch": 0.076,
"grad_norm": 106.0,
"learning_rate": 9.450000000000001e-06,
"loss": 19.9137,
"mean_token_accuracy": 0.6810434710234403,
"num_tokens": 940599.0,
"step": 190
},
{
"entropy": 1.256486358679831,
"epoch": 0.08,
"grad_norm": 117.5,
"learning_rate": 9.950000000000001e-06,
"loss": 20.0961,
"mean_token_accuracy": 0.6819845270365477,
"num_tokens": 989860.0,
"step": 200
},
{
"epoch": 0.08,
"eval_biology_entropy": 1.241776801943779,
"eval_biology_loss": 1.2506648302078247,
"eval_biology_mean_token_accuracy": 0.6855908244848251,
"eval_biology_num_tokens": 989860.0,
"eval_biology_runtime": 36.0247,
"eval_biology_samples_per_second": 13.879,
"eval_biology_steps_per_second": 13.879,
"step": 200
},
{
"epoch": 0.08,
"eval_chemistry_entropy": 0.9190568276047707,
"eval_chemistry_loss": 0.9248532652854919,
"eval_chemistry_mean_token_accuracy": 0.7533808985352516,
"eval_chemistry_num_tokens": 989860.0,
"eval_chemistry_runtime": 40.487,
"eval_chemistry_samples_per_second": 12.35,
"eval_chemistry_steps_per_second": 12.35,
"step": 200
},
{
"epoch": 0.08,
"eval_math_entropy": 0.7673128904104233,
"eval_math_loss": 0.9293037056922913,
"eval_math_mean_token_accuracy": 0.759642232298851,
"eval_math_num_tokens": 989860.0,
"eval_math_runtime": 41.5439,
"eval_math_samples_per_second": 12.035,
"eval_math_steps_per_second": 12.035,
"step": 200
},
{
"epoch": 0.08,
"eval_physics_entropy": 0.8287813304066658,
"eval_physics_loss": 0.8567265868186951,
"eval_physics_mean_token_accuracy": 0.770364968419075,
"eval_physics_num_tokens": 989860.0,
"eval_physics_runtime": 46.6277,
"eval_physics_samples_per_second": 10.723,
"eval_physics_steps_per_second": 10.723,
"step": 200
},
{
"entropy": 1.2084858063608408,
"epoch": 0.084,
"grad_norm": 99.5,
"learning_rate": 1.045e-05,
"loss": 19.469,
"mean_token_accuracy": 0.6908657155930996,
"num_tokens": 1041199.0,
"step": 210
},
{
"entropy": 1.3143941588699817,
"epoch": 0.088,
"grad_norm": 139.0,
"learning_rate": 1.095e-05,
"loss": 21.209,
"mean_token_accuracy": 0.670545787923038,
"num_tokens": 1087099.0,
"step": 220
},
{
"entropy": 1.2587614998221397,
"epoch": 0.092,
"grad_norm": 108.0,
"learning_rate": 1.145e-05,
"loss": 20.1256,
"mean_token_accuracy": 0.6842886801809073,
"num_tokens": 1135090.0,
"step": 230
},
{
"entropy": 1.2565513662993908,
"epoch": 0.096,
"grad_norm": 106.0,
"learning_rate": 1.195e-05,
"loss": 20.4044,
"mean_token_accuracy": 0.6776034630835056,
"num_tokens": 1181777.0,
"step": 240
},
{
"entropy": 1.2938017453998327,
"epoch": 0.1,
"grad_norm": 112.0,
"learning_rate": 1.2450000000000003e-05,
"loss": 20.3714,
"mean_token_accuracy": 0.6780621778219939,
"num_tokens": 1229242.0,
"step": 250
},
{
"entropy": 1.2320711638778448,
"epoch": 0.104,
"grad_norm": 94.5,
"learning_rate": 1.295e-05,
"loss": 20.052,
"mean_token_accuracy": 0.6799502000212669,
"num_tokens": 1283881.0,
"step": 260
},
{
"entropy": 1.2415869060903788,
"epoch": 0.108,
"grad_norm": 107.0,
"learning_rate": 1.3450000000000002e-05,
"loss": 19.9151,
"mean_token_accuracy": 0.6853171430528164,
"num_tokens": 1332507.0,
"step": 270
},
{
"entropy": 1.242128773033619,
"epoch": 0.112,
"grad_norm": 92.5,
"learning_rate": 1.3950000000000002e-05,
"loss": 19.9203,
"mean_token_accuracy": 0.685390866547823,
"num_tokens": 1385513.0,
"step": 280
},
{
"entropy": 1.2662223283201457,
"epoch": 0.116,
"grad_norm": 100.5,
"learning_rate": 1.4450000000000002e-05,
"loss": 20.3929,
"mean_token_accuracy": 0.6766710925847292,
"num_tokens": 1432251.0,
"step": 290
},
{
"entropy": 1.198459953069687,
"epoch": 0.12,
"grad_norm": 101.0,
"learning_rate": 1.4950000000000003e-05,
"loss": 19.3584,
"mean_token_accuracy": 0.68702756986022,
"num_tokens": 1483667.0,
"step": 300
},
{
"epoch": 0.12,
"eval_biology_entropy": 1.2469175053834916,
"eval_biology_loss": 1.2332121133804321,
"eval_biology_mean_token_accuracy": 0.6868851281404496,
"eval_biology_num_tokens": 1483667.0,
"eval_biology_runtime": 36.8445,
"eval_biology_samples_per_second": 13.571,
"eval_biology_steps_per_second": 13.571,
"step": 300
},
{
"epoch": 0.12,
"eval_chemistry_entropy": 0.9295264416337014,
"eval_chemistry_loss": 0.9073873162269592,
"eval_chemistry_mean_token_accuracy": 0.7570922123789787,
"eval_chemistry_num_tokens": 1483667.0,
"eval_chemistry_runtime": 40.4179,
"eval_chemistry_samples_per_second": 12.371,
"eval_chemistry_steps_per_second": 12.371,
"step": 300
},
{
"epoch": 0.12,
"eval_math_entropy": 0.7820550892949104,
"eval_math_loss": 0.9024779200553894,
"eval_math_mean_token_accuracy": 0.7648742213249207,
"eval_math_num_tokens": 1483667.0,
"eval_math_runtime": 41.3487,
"eval_math_samples_per_second": 12.092,
"eval_math_steps_per_second": 12.092,
"step": 300
},
{
"epoch": 0.12,
"eval_physics_entropy": 0.8381168316006661,
"eval_physics_loss": 0.8357300758361816,
"eval_physics_mean_token_accuracy": 0.7741321437358856,
"eval_physics_num_tokens": 1483667.0,
"eval_physics_runtime": 46.0597,
"eval_physics_samples_per_second": 10.855,
"eval_physics_steps_per_second": 10.855,
"step": 300
},
{
"entropy": 1.268108993396163,
"epoch": 0.124,
"grad_norm": 87.0,
"learning_rate": 1.545e-05,
"loss": 20.169,
"mean_token_accuracy": 0.6794606490060687,
"num_tokens": 1534286.0,
"step": 310
},
{
"entropy": 1.2664785664528608,
"epoch": 0.128,
"grad_norm": 95.0,
"learning_rate": 1.595e-05,
"loss": 20.1543,
"mean_token_accuracy": 0.6814448466524482,
"num_tokens": 1582368.0,
"step": 320
},
{
"entropy": 1.2443583220243455,
"epoch": 0.132,
"grad_norm": 100.5,
"learning_rate": 1.645e-05,
"loss": 20.1453,
"mean_token_accuracy": 0.6805377397686243,
"num_tokens": 1630878.0,
"step": 330
},
{
"entropy": 1.295349196344614,
"epoch": 0.136,
"grad_norm": 84.5,
"learning_rate": 1.6950000000000002e-05,
"loss": 20.5474,
"mean_token_accuracy": 0.6783911313861608,
"num_tokens": 1677060.0,
"step": 340
},
{
"entropy": 1.2136471576988697,
"epoch": 0.14,
"grad_norm": 96.0,
"learning_rate": 1.7450000000000004e-05,
"loss": 19.7364,
"mean_token_accuracy": 0.6849099691957236,
"num_tokens": 1725186.0,
"step": 350
},
{
"entropy": 1.2611235719174148,
"epoch": 0.144,
"grad_norm": 96.5,
"learning_rate": 1.795e-05,
"loss": 20.2599,
"mean_token_accuracy": 0.6779890628531575,
"num_tokens": 1773764.0,
"step": 360
},
{
"entropy": 1.24961077272892,
"epoch": 0.148,
"grad_norm": 93.0,
"learning_rate": 1.845e-05,
"loss": 19.9901,
"mean_token_accuracy": 0.6829751830548048,
"num_tokens": 1823706.0,
"step": 370
},
{
"entropy": 1.2856210090219975,
"epoch": 0.152,
"grad_norm": 103.5,
"learning_rate": 1.8950000000000003e-05,
"loss": 20.5301,
"mean_token_accuracy": 0.6794548545032739,
"num_tokens": 1873167.0,
"step": 380
},
{
"entropy": 1.28184983022511,
"epoch": 0.156,
"grad_norm": 87.5,
"learning_rate": 1.9450000000000002e-05,
"loss": 20.4718,
"mean_token_accuracy": 0.6774068776518106,
"num_tokens": 1921484.0,
"step": 390
},
{
"entropy": 1.2587715853005648,
"epoch": 0.16,
"grad_norm": 90.5,
"learning_rate": 1.9950000000000004e-05,
"loss": 20.2765,
"mean_token_accuracy": 0.6795550292357803,
"num_tokens": 1970077.0,
"step": 400
},
{
"epoch": 0.16,
"eval_biology_entropy": 1.2585812546014785,
"eval_biology_loss": 1.2385821342468262,
"eval_biology_mean_token_accuracy": 0.6883087210059166,
"eval_biology_num_tokens": 1970077.0,
"eval_biology_runtime": 40.6484,
"eval_biology_samples_per_second": 12.301,
"eval_biology_steps_per_second": 12.301,
"step": 400
},
{
"epoch": 0.16,
"eval_chemistry_entropy": 0.9248176854848862,
"eval_chemistry_loss": 0.9106165766716003,
"eval_chemistry_mean_token_accuracy": 0.7573499709963799,
"eval_chemistry_num_tokens": 1970077.0,
"eval_chemistry_runtime": 40.4652,
"eval_chemistry_samples_per_second": 12.356,
"eval_chemistry_steps_per_second": 12.356,
"step": 400
},
{
"epoch": 0.16,
"eval_math_entropy": 0.7678850111961365,
"eval_math_loss": 0.897013783454895,
"eval_math_mean_token_accuracy": 0.7673858489990234,
"eval_math_num_tokens": 1970077.0,
"eval_math_runtime": 41.3756,
"eval_math_samples_per_second": 12.084,
"eval_math_steps_per_second": 12.084,
"step": 400
},
{
"epoch": 0.16,
"eval_physics_entropy": 0.8230015668869018,
"eval_physics_loss": 0.8317809104919434,
"eval_physics_mean_token_accuracy": 0.7749597446918488,
"eval_physics_num_tokens": 1970077.0,
"eval_physics_runtime": 45.5271,
"eval_physics_samples_per_second": 10.982,
"eval_physics_steps_per_second": 10.982,
"step": 400
},
{
"entropy": 1.2881209932267665,
"epoch": 0.164,
"grad_norm": 80.5,
"learning_rate": 1.9950000000000004e-05,
"loss": 20.7614,
"mean_token_accuracy": 0.6718229403719306,
"num_tokens": 2019053.0,
"step": 410
},
{
"entropy": 1.2165789466351271,
"epoch": 0.168,
"grad_norm": 75.5,
"learning_rate": 1.9894444444444446e-05,
"loss": 19.5986,
"mean_token_accuracy": 0.6869289472699165,
"num_tokens": 2066815.0,
"step": 420
},
{
"entropy": 1.2494619188830256,
"epoch": 0.172,
"grad_norm": 80.5,
"learning_rate": 1.983888888888889e-05,
"loss": 20.2759,
"mean_token_accuracy": 0.6808055322617292,
"num_tokens": 2116402.0,
"step": 430
},
{
"entropy": 1.214642644301057,
"epoch": 0.176,
"grad_norm": 81.5,
"learning_rate": 1.9783333333333334e-05,
"loss": 19.3769,
"mean_token_accuracy": 0.6881225043907762,
"num_tokens": 2168354.0,
"step": 440
},
{
"entropy": 1.2326878625899553,
"epoch": 0.18,
"grad_norm": 76.0,
"learning_rate": 1.972777777777778e-05,
"loss": 20.0507,
"mean_token_accuracy": 0.6823928710073233,
"num_tokens": 2219158.0,
"step": 450
},
{
"entropy": 1.2790402939543128,
"epoch": 0.184,
"grad_norm": 97.0,
"learning_rate": 1.9672222222222222e-05,
"loss": 20.4576,
"mean_token_accuracy": 0.6788654251024127,
"num_tokens": 2270617.0,
"step": 460
},
{
"entropy": 1.2233176242560149,
"epoch": 0.188,
"grad_norm": 76.0,
"learning_rate": 1.9616666666666668e-05,
"loss": 19.4471,
"mean_token_accuracy": 0.6852962577715516,
"num_tokens": 2322034.0,
"step": 470
},
{
"entropy": 1.284807311743498,
"epoch": 0.192,
"grad_norm": 97.0,
"learning_rate": 1.9561111111111114e-05,
"loss": 21.2325,
"mean_token_accuracy": 0.6681901682168245,
"num_tokens": 2365822.0,
"step": 480
},
{
"entropy": 1.250083539262414,
"epoch": 0.196,
"grad_norm": 93.5,
"learning_rate": 1.9505555555555556e-05,
"loss": 19.6203,
"mean_token_accuracy": 0.6853536464273929,
"num_tokens": 2413853.0,
"step": 490
},
{
"entropy": 1.2981292121112347,
"epoch": 0.2,
"grad_norm": 85.0,
"learning_rate": 1.9450000000000002e-05,
"loss": 20.8012,
"mean_token_accuracy": 0.6707905715331435,
"num_tokens": 2459876.0,
"step": 500
},
{
"epoch": 0.2,
"eval_biology_entropy": 1.2485336683392525,
"eval_biology_loss": 1.2281557321548462,
"eval_biology_mean_token_accuracy": 0.6897270123958588,
"eval_biology_num_tokens": 2459876.0,
"eval_biology_runtime": 38.8945,
"eval_biology_samples_per_second": 12.855,
"eval_biology_steps_per_second": 12.855,
"step": 500
},
{
"epoch": 0.2,
"eval_chemistry_entropy": 0.9259556359648704,
"eval_chemistry_loss": 0.9128974676132202,
"eval_chemistry_mean_token_accuracy": 0.7565669685602188,
"eval_chemistry_num_tokens": 2459876.0,
"eval_chemistry_runtime": 40.516,
"eval_chemistry_samples_per_second": 12.341,
"eval_chemistry_steps_per_second": 12.341,
"step": 500
},
{
"epoch": 0.2,
"eval_math_entropy": 0.7720121605396271,
"eval_math_loss": 0.9020798206329346,
"eval_math_mean_token_accuracy": 0.7660462435483932,
"eval_math_num_tokens": 2459876.0,
"eval_math_runtime": 62.8982,
"eval_math_samples_per_second": 7.949,
"eval_math_steps_per_second": 7.949,
"step": 500
},
{
"epoch": 0.2,
"eval_physics_entropy": 0.8241469451785087,
"eval_physics_loss": 0.832719624042511,
"eval_physics_mean_token_accuracy": 0.7746937532424927,
"eval_physics_num_tokens": 2459876.0,
"eval_physics_runtime": 76.8817,
"eval_physics_samples_per_second": 6.504,
"eval_physics_steps_per_second": 6.504,
"step": 500
}
],
"logging_steps": 10,
"max_steps": 4000,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"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.6494542652385037e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}