Microscopic-Olmo-2B-3.9k-steps / trainer_state.json
DrNicefellow's picture
Upload 13 files
dbb6e74 verified
Raw
History Blame Contribute Delete
23.3 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.03178549194008557,
"eval_steps": 100000000,
"global_step": 3900,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 8.15012613848348e-06,
"grad_norm": 3.4710192680358887,
"learning_rate": 1.0000000000000001e-07,
"loss": 11.2605,
"step": 1
},
{
"epoch": 0.0002445037841545044,
"grad_norm": 5.464046955108643,
"learning_rate": 3e-06,
"loss": 10.9977,
"step": 30
},
{
"epoch": 0.0004890075683090088,
"grad_norm": 1.1376756429672241,
"learning_rate": 6e-06,
"loss": 9.4082,
"step": 60
},
{
"epoch": 0.0007335113524635132,
"grad_norm": 1.2878344058990479,
"learning_rate": 9e-06,
"loss": 8.549,
"step": 90
},
{
"epoch": 0.0009780151366180175,
"grad_norm": 1.2800588607788086,
"learning_rate": 1.2e-05,
"loss": 8.0361,
"step": 120
},
{
"epoch": 0.001222518920772522,
"grad_norm": 1.4315314292907715,
"learning_rate": 1.5e-05,
"loss": 7.653,
"step": 150
},
{
"epoch": 0.0014670227049270264,
"grad_norm": 1.381317377090454,
"learning_rate": 1.8e-05,
"loss": 7.4179,
"step": 180
},
{
"epoch": 0.0017115264890815308,
"grad_norm": 1.7989460229873657,
"learning_rate": 2.1e-05,
"loss": 7.2012,
"step": 210
},
{
"epoch": 0.001956030273236035,
"grad_norm": 1.3431414365768433,
"learning_rate": 2.4e-05,
"loss": 7.0383,
"step": 240
},
{
"epoch": 0.0022005340573905395,
"grad_norm": 1.028826117515564,
"learning_rate": 2.7000000000000002e-05,
"loss": 6.866,
"step": 270
},
{
"epoch": 0.002445037841545044,
"grad_norm": 1.201025128364563,
"learning_rate": 3e-05,
"loss": 6.717,
"step": 300
},
{
"epoch": 0.0026895416256995483,
"grad_norm": 1.1023098230361938,
"learning_rate": 3.3e-05,
"loss": 6.5535,
"step": 330
},
{
"epoch": 0.0029340454098540527,
"grad_norm": 1.2839674949645996,
"learning_rate": 3.6e-05,
"loss": 6.4022,
"step": 360
},
{
"epoch": 0.003178549194008557,
"grad_norm": 2.267265796661377,
"learning_rate": 3.9000000000000006e-05,
"loss": 6.2858,
"step": 390
},
{
"epoch": 0.0034230529781630616,
"grad_norm": 1.0635628700256348,
"learning_rate": 4.2e-05,
"loss": 6.1681,
"step": 420
},
{
"epoch": 0.0036675567623175656,
"grad_norm": 1.263838768005371,
"learning_rate": 4.5e-05,
"loss": 6.0728,
"step": 450
},
{
"epoch": 0.00391206054647207,
"grad_norm": 1.4611454010009766,
"learning_rate": 4.8e-05,
"loss": 5.972,
"step": 480
},
{
"epoch": 0.0041565643306265745,
"grad_norm": 1.0120561122894287,
"learning_rate": 4.999999990869806e-05,
"loss": 5.8619,
"step": 510
},
{
"epoch": 0.004401068114781079,
"grad_norm": 1.1349974870681763,
"learning_rate": 4.999999853916893e-05,
"loss": 5.785,
"step": 540
},
{
"epoch": 0.004645571898935583,
"grad_norm": 1.0840613842010498,
"learning_rate": 4.9999995526204936e-05,
"loss": 5.7071,
"step": 570
},
{
"epoch": 0.004890075683090088,
"grad_norm": 1.258074402809143,
"learning_rate": 4.999999086980628e-05,
"loss": 5.6199,
"step": 600
},
{
"epoch": 0.005134579467244592,
"grad_norm": 1.284726858139038,
"learning_rate": 4.999998456997326e-05,
"loss": 5.5465,
"step": 630
},
{
"epoch": 0.005379083251399097,
"grad_norm": 1.2079874277114868,
"learning_rate": 4.999997662670628e-05,
"loss": 5.4816,
"step": 660
},
{
"epoch": 0.005623587035553601,
"grad_norm": 1.3364052772521973,
"learning_rate": 4.999996704000589e-05,
"loss": 5.4079,
"step": 690
},
{
"epoch": 0.0058680908197081055,
"grad_norm": 0.9860705137252808,
"learning_rate": 4.99999558098727e-05,
"loss": 5.3598,
"step": 720
},
{
"epoch": 0.00611259460386261,
"grad_norm": 1.2071930170059204,
"learning_rate": 4.9999942936307445e-05,
"loss": 5.2884,
"step": 750
},
{
"epoch": 0.006357098388017114,
"grad_norm": 0.8959563970565796,
"learning_rate": 4.9999928419310994e-05,
"loss": 5.2391,
"step": 780
},
{
"epoch": 0.006601602172171619,
"grad_norm": 1.2356096506118774,
"learning_rate": 4.999991225888427e-05,
"loss": 5.1879,
"step": 810
},
{
"epoch": 0.006846105956326123,
"grad_norm": 0.9705113172531128,
"learning_rate": 4.999989445502837e-05,
"loss": 5.1424,
"step": 840
},
{
"epoch": 0.007090609740480627,
"grad_norm": 0.9504437446594238,
"learning_rate": 4.9999875007744436e-05,
"loss": 5.0966,
"step": 870
},
{
"epoch": 0.007335113524635131,
"grad_norm": 0.9488673806190491,
"learning_rate": 4.9999853917033756e-05,
"loss": 5.0424,
"step": 900
},
{
"epoch": 0.007579617308789636,
"grad_norm": 0.959373950958252,
"learning_rate": 4.999983118289773e-05,
"loss": 5.0387,
"step": 930
},
{
"epoch": 0.00782412109294414,
"grad_norm": 0.8465414643287659,
"learning_rate": 4.999980680533782e-05,
"loss": 4.9769,
"step": 960
},
{
"epoch": 0.008068624877098645,
"grad_norm": 0.8328993916511536,
"learning_rate": 4.999978078435567e-05,
"loss": 4.9335,
"step": 990
},
{
"epoch": 0.008313128661253149,
"grad_norm": 0.8107655644416809,
"learning_rate": 4.999975311995295e-05,
"loss": 4.9214,
"step": 1020
},
{
"epoch": 0.008557632445407654,
"grad_norm": 0.8149654865264893,
"learning_rate": 4.99997238121315e-05,
"loss": 4.8651,
"step": 1050
},
{
"epoch": 0.008802136229562158,
"grad_norm": 0.8837414979934692,
"learning_rate": 4.999969286089325e-05,
"loss": 4.8327,
"step": 1080
},
{
"epoch": 0.009046640013716663,
"grad_norm": 1.1360137462615967,
"learning_rate": 4.9999660266240235e-05,
"loss": 4.7906,
"step": 1110
},
{
"epoch": 0.009291143797871167,
"grad_norm": 0.7992026209831238,
"learning_rate": 4.9999626028174585e-05,
"loss": 4.7612,
"step": 1140
},
{
"epoch": 0.00953564758202567,
"grad_norm": 0.8481825590133667,
"learning_rate": 4.999959014669856e-05,
"loss": 4.7106,
"step": 1170
},
{
"epoch": 0.009780151366180176,
"grad_norm": 0.8183879256248474,
"learning_rate": 4.9999552621814513e-05,
"loss": 4.6993,
"step": 1200
},
{
"epoch": 0.010024655150334679,
"grad_norm": 0.8460689187049866,
"learning_rate": 4.9999513453524917e-05,
"loss": 4.6664,
"step": 1230
},
{
"epoch": 0.010269158934489184,
"grad_norm": 0.8723706007003784,
"learning_rate": 4.9999472641832336e-05,
"loss": 4.6371,
"step": 1260
},
{
"epoch": 0.010513662718643688,
"grad_norm": 0.7682787179946899,
"learning_rate": 4.999943018673946e-05,
"loss": 4.6184,
"step": 1290
},
{
"epoch": 0.010758166502798193,
"grad_norm": 0.9049955606460571,
"learning_rate": 4.999938608824909e-05,
"loss": 4.5968,
"step": 1320
},
{
"epoch": 0.011002670286952697,
"grad_norm": 0.7860899567604065,
"learning_rate": 4.999934034636411e-05,
"loss": 4.5266,
"step": 1350
},
{
"epoch": 0.011247174071107202,
"grad_norm": 0.7918768525123596,
"learning_rate": 4.999929296108753e-05,
"loss": 4.5069,
"step": 1380
},
{
"epoch": 0.011491677855261706,
"grad_norm": 0.6713089942932129,
"learning_rate": 4.9999243932422466e-05,
"loss": 4.4663,
"step": 1410
},
{
"epoch": 0.011736181639416211,
"grad_norm": 0.756101131439209,
"learning_rate": 4.999919326037215e-05,
"loss": 4.4782,
"step": 1440
},
{
"epoch": 0.011980685423570715,
"grad_norm": 0.7067996263504028,
"learning_rate": 4.99991409449399e-05,
"loss": 4.4388,
"step": 1470
},
{
"epoch": 0.01222518920772522,
"grad_norm": 0.7132194638252258,
"learning_rate": 4.999908698612916e-05,
"loss": 4.4135,
"step": 1500
},
{
"epoch": 0.012469692991879723,
"grad_norm": 0.7282501459121704,
"learning_rate": 4.9999031383943486e-05,
"loss": 4.4057,
"step": 1530
},
{
"epoch": 0.012714196776034229,
"grad_norm": 0.6851722598075867,
"learning_rate": 4.999897413838651e-05,
"loss": 4.3612,
"step": 1560
},
{
"epoch": 0.012958700560188732,
"grad_norm": 0.7953110337257385,
"learning_rate": 4.999891524946202e-05,
"loss": 4.3268,
"step": 1590
},
{
"epoch": 0.013203204344343238,
"grad_norm": 0.726719319820404,
"learning_rate": 4.999885471717387e-05,
"loss": 4.314,
"step": 1620
},
{
"epoch": 0.013447708128497741,
"grad_norm": 0.7354792356491089,
"learning_rate": 4.999879254152605e-05,
"loss": 4.2807,
"step": 1650
},
{
"epoch": 0.013692211912652246,
"grad_norm": 0.7072017788887024,
"learning_rate": 4.999872872252265e-05,
"loss": 4.282,
"step": 1680
},
{
"epoch": 0.01393671569680675,
"grad_norm": 0.690112292766571,
"learning_rate": 4.999866326016785e-05,
"loss": 4.2686,
"step": 1710
},
{
"epoch": 0.014181219480961254,
"grad_norm": 0.697634756565094,
"learning_rate": 4.999859615446596e-05,
"loss": 4.2662,
"step": 1740
},
{
"epoch": 0.014425723265115759,
"grad_norm": 0.6999862790107727,
"learning_rate": 4.99985274054214e-05,
"loss": 4.2207,
"step": 1770
},
{
"epoch": 0.014670227049270262,
"grad_norm": 0.6920334100723267,
"learning_rate": 4.999845701303868e-05,
"loss": 4.2163,
"step": 1800
},
{
"epoch": 0.014914730833424768,
"grad_norm": 0.6884493827819824,
"learning_rate": 4.999838497732243e-05,
"loss": 4.209,
"step": 1830
},
{
"epoch": 0.015159234617579271,
"grad_norm": 0.776447057723999,
"learning_rate": 4.999831129827739e-05,
"loss": 4.1856,
"step": 1860
},
{
"epoch": 0.015403738401733777,
"grad_norm": 0.7332949042320251,
"learning_rate": 4.9998235975908394e-05,
"loss": 4.156,
"step": 1890
},
{
"epoch": 0.01564824218588828,
"grad_norm": 0.6691558361053467,
"learning_rate": 4.99981590102204e-05,
"loss": 4.1339,
"step": 1920
},
{
"epoch": 0.015892745970042785,
"grad_norm": 0.6789201498031616,
"learning_rate": 4.9998080401218464e-05,
"loss": 4.1344,
"step": 1950
},
{
"epoch": 0.01613724975419729,
"grad_norm": 0.6654175519943237,
"learning_rate": 4.999800014890777e-05,
"loss": 4.1295,
"step": 1980
},
{
"epoch": 0.016381753538351793,
"grad_norm": 0.6632519364356995,
"learning_rate": 4.9997918253293555e-05,
"loss": 4.1036,
"step": 2010
},
{
"epoch": 0.016626257322506298,
"grad_norm": 0.6618740558624268,
"learning_rate": 4.999783471438124e-05,
"loss": 4.0857,
"step": 2040
},
{
"epoch": 0.016870761106660803,
"grad_norm": 0.6384952068328857,
"learning_rate": 4.999774953217631e-05,
"loss": 4.0755,
"step": 2070
},
{
"epoch": 0.01711526489081531,
"grad_norm": 0.6808215975761414,
"learning_rate": 4.9997662706684345e-05,
"loss": 4.0568,
"step": 2100
},
{
"epoch": 0.01735976867496981,
"grad_norm": 0.6514068841934204,
"learning_rate": 4.999757423791107e-05,
"loss": 4.0474,
"step": 2130
},
{
"epoch": 0.017604272459124316,
"grad_norm": 0.6691142320632935,
"learning_rate": 4.9997484125862306e-05,
"loss": 4.0467,
"step": 2160
},
{
"epoch": 0.01784877624327882,
"grad_norm": 0.6240984201431274,
"learning_rate": 4.999739237054395e-05,
"loss": 4.0259,
"step": 2190
},
{
"epoch": 0.018093280027433326,
"grad_norm": 0.6286123991012573,
"learning_rate": 4.9997298971962065e-05,
"loss": 4.0098,
"step": 2220
},
{
"epoch": 0.018337783811587828,
"grad_norm": 0.6232910752296448,
"learning_rate": 4.999720393012277e-05,
"loss": 4.0203,
"step": 2250
},
{
"epoch": 0.018582287595742333,
"grad_norm": 0.6792376041412354,
"learning_rate": 4.999710724503233e-05,
"loss": 3.9909,
"step": 2280
},
{
"epoch": 0.01882679137989684,
"grad_norm": 0.6824718117713928,
"learning_rate": 4.9997008916697075e-05,
"loss": 3.9922,
"step": 2310
},
{
"epoch": 0.01907129516405134,
"grad_norm": 0.6520410776138306,
"learning_rate": 4.999690894512349e-05,
"loss": 3.9533,
"step": 2340
},
{
"epoch": 0.019315798948205846,
"grad_norm": 0.6467030048370361,
"learning_rate": 4.999680733031814e-05,
"loss": 3.9464,
"step": 2370
},
{
"epoch": 0.01956030273236035,
"grad_norm": 0.6356080770492554,
"learning_rate": 4.9996704072287716e-05,
"loss": 3.9288,
"step": 2400
},
{
"epoch": 0.019804806516514856,
"grad_norm": 0.6498362421989441,
"learning_rate": 4.9996599171038984e-05,
"loss": 3.9345,
"step": 2430
},
{
"epoch": 0.020049310300669358,
"grad_norm": 0.6107562184333801,
"learning_rate": 4.999649262657886e-05,
"loss": 3.9368,
"step": 2460
},
{
"epoch": 0.020293814084823864,
"grad_norm": 0.6478250622749329,
"learning_rate": 4.999638443891434e-05,
"loss": 3.9273,
"step": 2490
},
{
"epoch": 0.02053831786897837,
"grad_norm": 0.631263256072998,
"learning_rate": 4.999627460805253e-05,
"loss": 3.9136,
"step": 2520
},
{
"epoch": 0.020782821653132874,
"grad_norm": 0.6120920777320862,
"learning_rate": 4.999616313400066e-05,
"loss": 3.8834,
"step": 2550
},
{
"epoch": 0.021027325437287376,
"grad_norm": 0.6151197552680969,
"learning_rate": 4.999605001676605e-05,
"loss": 3.8994,
"step": 2580
},
{
"epoch": 0.02127182922144188,
"grad_norm": 0.623715877532959,
"learning_rate": 4.9995935256356144e-05,
"loss": 3.8929,
"step": 2610
},
{
"epoch": 0.021516333005596387,
"grad_norm": 0.6688769459724426,
"learning_rate": 4.9995818852778476e-05,
"loss": 3.8499,
"step": 2640
},
{
"epoch": 0.021760836789750892,
"grad_norm": 0.6272155046463013,
"learning_rate": 4.999570080604071e-05,
"loss": 3.8861,
"step": 2670
},
{
"epoch": 0.022005340573905394,
"grad_norm": 0.597653329372406,
"learning_rate": 4.99955811161506e-05,
"loss": 3.8674,
"step": 2700
},
{
"epoch": 0.0222498443580599,
"grad_norm": 0.5608483552932739,
"learning_rate": 4.9995459783116004e-05,
"loss": 3.8493,
"step": 2730
},
{
"epoch": 0.022494348142214404,
"grad_norm": 0.5991063117980957,
"learning_rate": 4.999533680694493e-05,
"loss": 3.8454,
"step": 2760
},
{
"epoch": 0.02273885192636891,
"grad_norm": 0.5738102793693542,
"learning_rate": 4.9995212187645416e-05,
"loss": 3.8395,
"step": 2790
},
{
"epoch": 0.02298335571052341,
"grad_norm": 0.6145568490028381,
"learning_rate": 4.9995085925225693e-05,
"loss": 3.8313,
"step": 2820
},
{
"epoch": 0.023227859494677917,
"grad_norm": 0.6019515991210938,
"learning_rate": 4.999495801969404e-05,
"loss": 3.8277,
"step": 2850
},
{
"epoch": 0.023472363278832422,
"grad_norm": 0.6177758574485779,
"learning_rate": 4.9994828471058876e-05,
"loss": 3.8083,
"step": 2880
},
{
"epoch": 0.023716867062986924,
"grad_norm": 0.5616528987884521,
"learning_rate": 4.9994697279328714e-05,
"loss": 3.8114,
"step": 2910
},
{
"epoch": 0.02396137084714143,
"grad_norm": 0.6194447875022888,
"learning_rate": 4.9994564444512176e-05,
"loss": 3.8183,
"step": 2940
},
{
"epoch": 0.024205874631295934,
"grad_norm": 0.578895092010498,
"learning_rate": 4.9994429966618e-05,
"loss": 3.7871,
"step": 2970
},
{
"epoch": 0.02445037841545044,
"grad_norm": 0.6014060378074646,
"learning_rate": 4.999429384565502e-05,
"loss": 3.7711,
"step": 3000
},
{
"epoch": 0.02469488219960494,
"grad_norm": 0.5589067339897156,
"learning_rate": 4.999415608163217e-05,
"loss": 3.7533,
"step": 3030
},
{
"epoch": 0.024939385983759447,
"grad_norm": 0.5722873210906982,
"learning_rate": 4.999401667455854e-05,
"loss": 3.7585,
"step": 3060
},
{
"epoch": 0.025183889767913952,
"grad_norm": 0.5734965801239014,
"learning_rate": 4.9993875624443274e-05,
"loss": 3.77,
"step": 3090
},
{
"epoch": 0.025428393552068457,
"grad_norm": 0.5524207353591919,
"learning_rate": 4.9993732931295646e-05,
"loss": 3.718,
"step": 3120
},
{
"epoch": 0.02567289733622296,
"grad_norm": 0.5776082277297974,
"learning_rate": 4.999358859512503e-05,
"loss": 3.7573,
"step": 3150
},
{
"epoch": 0.025917401120377465,
"grad_norm": 0.5799595713615417,
"learning_rate": 4.9993442615940936e-05,
"loss": 3.7552,
"step": 3180
},
{
"epoch": 0.02616190490453197,
"grad_norm": 0.5820346474647522,
"learning_rate": 4.999329499375292e-05,
"loss": 3.7394,
"step": 3210
},
{
"epoch": 0.026406408688686475,
"grad_norm": 0.5450282096862793,
"learning_rate": 4.999314572857074e-05,
"loss": 3.7393,
"step": 3240
},
{
"epoch": 0.026650912472840977,
"grad_norm": 0.6213249564170837,
"learning_rate": 4.9992994820404174e-05,
"loss": 3.7191,
"step": 3270
},
{
"epoch": 0.026895416256995482,
"grad_norm": 0.5940688848495483,
"learning_rate": 4.999284226926314e-05,
"loss": 3.719,
"step": 3300
},
{
"epoch": 0.027139920041149988,
"grad_norm": 0.5779993534088135,
"learning_rate": 4.999268807515768e-05,
"loss": 3.7092,
"step": 3330
},
{
"epoch": 0.027384423825304493,
"grad_norm": 0.5899255871772766,
"learning_rate": 4.999253223809792e-05,
"loss": 3.6939,
"step": 3360
},
{
"epoch": 0.027628927609458995,
"grad_norm": 0.6330375671386719,
"learning_rate": 4.999237475809411e-05,
"loss": 3.7102,
"step": 3390
},
{
"epoch": 0.0278734313936135,
"grad_norm": 0.5771914720535278,
"learning_rate": 4.99922156351566e-05,
"loss": 3.7196,
"step": 3420
},
{
"epoch": 0.028117935177768005,
"grad_norm": 0.5769143104553223,
"learning_rate": 4.999205486929586e-05,
"loss": 3.7127,
"step": 3450
},
{
"epoch": 0.028362438961922507,
"grad_norm": 0.5581954121589661,
"learning_rate": 4.999189246052245e-05,
"loss": 3.689,
"step": 3480
},
{
"epoch": 0.028606942746077012,
"grad_norm": 0.6041043400764465,
"learning_rate": 4.999172840884704e-05,
"loss": 3.6831,
"step": 3510
},
{
"epoch": 0.028851446530231518,
"grad_norm": 0.5458335280418396,
"learning_rate": 4.999156271428043e-05,
"loss": 3.6694,
"step": 3540
},
{
"epoch": 0.029095950314386023,
"grad_norm": 0.5470607280731201,
"learning_rate": 4.9991395376833496e-05,
"loss": 3.6702,
"step": 3570
},
{
"epoch": 0.029340454098540525,
"grad_norm": 0.5878787040710449,
"learning_rate": 4.999122639651725e-05,
"loss": 3.6492,
"step": 3600
},
{
"epoch": 0.02958495788269503,
"grad_norm": 0.5691691637039185,
"learning_rate": 4.9991055773342795e-05,
"loss": 3.6812,
"step": 3630
},
{
"epoch": 0.029829461666849535,
"grad_norm": 0.5548356771469116,
"learning_rate": 4.9990883507321354e-05,
"loss": 3.645,
"step": 3660
},
{
"epoch": 0.03007396545100404,
"grad_norm": 0.5686156153678894,
"learning_rate": 4.999070959846424e-05,
"loss": 3.6505,
"step": 3690
},
{
"epoch": 0.030318469235158543,
"grad_norm": 0.5596534013748169,
"learning_rate": 4.999053404678289e-05,
"loss": 3.6532,
"step": 3720
},
{
"epoch": 0.030562973019313048,
"grad_norm": 0.54421067237854,
"learning_rate": 4.999035685228884e-05,
"loss": 3.6238,
"step": 3750
},
{
"epoch": 0.030807476803467553,
"grad_norm": 0.56732177734375,
"learning_rate": 4.999017801499375e-05,
"loss": 3.6449,
"step": 3780
},
{
"epoch": 0.03105198058762206,
"grad_norm": 0.5608410239219666,
"learning_rate": 4.998999753490937e-05,
"loss": 3.6344,
"step": 3810
},
{
"epoch": 0.03129648437177656,
"grad_norm": 0.5919491648674011,
"learning_rate": 4.998981541204757e-05,
"loss": 3.6213,
"step": 3840
},
{
"epoch": 0.031540988155931066,
"grad_norm": 0.5795233249664307,
"learning_rate": 4.998963164642031e-05,
"loss": 3.6239,
"step": 3870
},
{
"epoch": 0.03178549194008557,
"grad_norm": 0.5669205784797668,
"learning_rate": 4.9989446238039676e-05,
"loss": 3.5831,
"step": 3900
}
],
"logging_steps": 30,
"max_steps": 368091,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 100,
"total_flos": 3.730642953043968e+18,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}