ScriptSwiftAIV2 / checkpoint-1000 /trainer_state.json
Dikshan1234's picture
checkpoint-1000 | step 1000 | loss 0.4174 | 2.5% complete
f7d1cc1 verified
Raw
History Blame Contribute Delete
19.5 kB
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.028175755040453325,
"eval_steps": 500,
"global_step": 1000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0002922796166022129,
"grad_norm": 0.3604201674461365,
"learning_rate": 1.8e-06,
"loss": 0.6163836479187011,
"step": 10
},
{
"epoch": 0.0005845592332044258,
"grad_norm": 0.307494193315506,
"learning_rate": 3.8e-06,
"loss": 0.6056640148162842,
"step": 20
},
{
"epoch": 0.00014613980830110646,
"grad_norm": 0.1861964464187622,
"learning_rate": 5.8e-06,
"loss": 0.5507771968841553,
"step": 30
},
{
"epoch": 0.00011691184664088517,
"grad_norm": 0.18063770234584808,
"learning_rate": 7.8e-06,
"loss": 0.5524736642837524,
"step": 40
},
{
"epoch": 0.0004091914632430981,
"grad_norm": 0.1831495314836502,
"learning_rate": 9.800000000000001e-06,
"loss": 0.5550893783569336,
"step": 50
},
{
"epoch": 0.000701471079845311,
"grad_norm": 0.2708336412906647,
"learning_rate": 1.18e-05,
"loss": 0.5027066707611084,
"step": 60
},
{
"epoch": 0.0009937506964475238,
"grad_norm": 0.16506414115428925,
"learning_rate": 1.3800000000000002e-05,
"loss": 0.5025710105895996,
"step": 70
},
{
"epoch": 0.0012860303130497368,
"grad_norm": 0.7222521305084229,
"learning_rate": 1.58e-05,
"loss": 0.44574794769287107,
"step": 80
},
{
"epoch": 0.0015783099296519498,
"grad_norm": 0.3094443678855896,
"learning_rate": 1.78e-05,
"loss": 0.463105583190918,
"step": 90
},
{
"epoch": 0.0018705895462541628,
"grad_norm": 0.22881925106048584,
"learning_rate": 1.9800000000000004e-05,
"loss": 0.47495498657226565,
"step": 100
},
{
"epoch": 0.0021628691628563755,
"grad_norm": 0.2994537055492401,
"learning_rate": 2.18e-05,
"loss": 0.44906439781188967,
"step": 110
},
{
"epoch": 0.0024551487794585887,
"grad_norm": 0.22821807861328125,
"learning_rate": 2.38e-05,
"loss": 0.4511695384979248,
"step": 120
},
{
"epoch": 0.0027474283960608015,
"grad_norm": 0.23172526061534882,
"learning_rate": 2.58e-05,
"loss": 0.45236430168151853,
"step": 130
},
{
"epoch": 0.003039708012663014,
"grad_norm": 0.25378280878067017,
"learning_rate": 2.7800000000000005e-05,
"loss": 0.49311389923095705,
"step": 140
},
{
"epoch": 0.0033319876292652274,
"grad_norm": 0.30715328454971313,
"learning_rate": 2.98e-05,
"loss": 0.45659761428833007,
"step": 150
},
{
"epoch": 0.00362426724586744,
"grad_norm": 0.2318553328514099,
"learning_rate": 3.18e-05,
"loss": 0.48353352546691897,
"step": 160
},
{
"epoch": 0.003916546862469653,
"grad_norm": 0.3422481417655945,
"learning_rate": 3.38e-05,
"loss": 0.48243284225463867,
"step": 170
},
{
"epoch": 0.004208826479071866,
"grad_norm": 0.2627033591270447,
"learning_rate": 3.58e-05,
"loss": 0.44968676567077637,
"step": 180
},
{
"epoch": 0.004501106095674079,
"grad_norm": 0.270560622215271,
"learning_rate": 3.7800000000000004e-05,
"loss": 0.43050732612609866,
"step": 190
},
{
"epoch": 0.004793385712276292,
"grad_norm": 0.29916077852249146,
"learning_rate": 3.9800000000000005e-05,
"loss": 0.38156795501708984,
"step": 200
},
{
"epoch": 0.005085665328878505,
"grad_norm": 0.29618600010871887,
"learning_rate": 4.18e-05,
"loss": 0.41876654624938964,
"step": 210
},
{
"epoch": 0.005377944945480718,
"grad_norm": 0.24618731439113617,
"learning_rate": 4.38e-05,
"loss": 0.4200754642486572,
"step": 220
},
{
"epoch": 0.00567022456208293,
"grad_norm": 0.4028846323490143,
"learning_rate": 4.58e-05,
"loss": 0.4128837585449219,
"step": 230
},
{
"epoch": 0.0059625041786851435,
"grad_norm": 0.2899196445941925,
"learning_rate": 4.78e-05,
"loss": 0.4727485656738281,
"step": 240
},
{
"epoch": 0.006254783795287357,
"grad_norm": 0.372092068195343,
"learning_rate": 4.9800000000000004e-05,
"loss": 0.3837599754333496,
"step": 250
},
{
"epoch": 0.00654706341188957,
"grad_norm": 0.27367591857910156,
"learning_rate": 5.1800000000000005e-05,
"loss": 0.4533690929412842,
"step": 260
},
{
"epoch": 0.006839343028491782,
"grad_norm": 0.2826191782951355,
"learning_rate": 5.380000000000001e-05,
"loss": 0.5056231021881104,
"step": 270
},
{
"epoch": 0.007131622645093995,
"grad_norm": 0.34035593271255493,
"learning_rate": 5.580000000000001e-05,
"loss": 0.4258269309997559,
"step": 280
},
{
"epoch": 0.0074239022616962086,
"grad_norm": 0.2767459750175476,
"learning_rate": 5.7799999999999995e-05,
"loss": 0.3629124641418457,
"step": 290
},
{
"epoch": 0.007716181878298421,
"grad_norm": 0.2805218994617462,
"learning_rate": 5.9800000000000003e-05,
"loss": 0.35598738193511964,
"step": 300
},
{
"epoch": 0.008008461494900634,
"grad_norm": 0.32115164399147034,
"learning_rate": 6.18e-05,
"loss": 0.45623183250427246,
"step": 310
},
{
"epoch": 0.008300741111502847,
"grad_norm": 0.25464335083961487,
"learning_rate": 6.38e-05,
"loss": 0.3766467809677124,
"step": 320
},
{
"epoch": 0.00859302072810506,
"grad_norm": 0.240229070186615,
"learning_rate": 6.58e-05,
"loss": 0.4206503391265869,
"step": 330
},
{
"epoch": 0.008885300344707274,
"grad_norm": 0.24640053510665894,
"learning_rate": 6.780000000000001e-05,
"loss": 0.42534866333007815,
"step": 340
},
{
"epoch": 0.009177579961309485,
"grad_norm": 0.22595764696598053,
"learning_rate": 6.98e-05,
"loss": 0.4169299602508545,
"step": 350
},
{
"epoch": 0.009469859577911698,
"grad_norm": 0.2035064995288849,
"learning_rate": 7.18e-05,
"loss": 0.36454784870147705,
"step": 360
},
{
"epoch": 0.009762139194513911,
"grad_norm": 0.21908830106258392,
"learning_rate": 7.38e-05,
"loss": 0.39443838596343994,
"step": 370
},
{
"epoch": 0.010054418811116125,
"grad_norm": 0.2039889395236969,
"learning_rate": 7.58e-05,
"loss": 0.4549224853515625,
"step": 380
},
{
"epoch": 0.010346698427718338,
"grad_norm": 0.24606066942214966,
"learning_rate": 7.780000000000001e-05,
"loss": 0.3763291835784912,
"step": 390
},
{
"epoch": 0.010638978044320551,
"grad_norm": 0.4349507689476013,
"learning_rate": 7.98e-05,
"loss": 0.42160859107971194,
"step": 400
},
{
"epoch": 0.010931257660922762,
"grad_norm": 0.2086913138628006,
"learning_rate": 8.18e-05,
"loss": 0.4335451126098633,
"step": 410
},
{
"epoch": 0.011223537277524976,
"grad_norm": 0.21672461926937103,
"learning_rate": 8.38e-05,
"loss": 0.39824934005737306,
"step": 420
},
{
"epoch": 0.011515816894127189,
"grad_norm": 0.34380897879600525,
"learning_rate": 8.58e-05,
"loss": 0.4544211387634277,
"step": 430
},
{
"epoch": 0.011808096510729402,
"grad_norm": 0.2024194449186325,
"learning_rate": 8.78e-05,
"loss": 0.4060193061828613,
"step": 440
},
{
"epoch": 0.012100376127331615,
"grad_norm": 0.22563715279102325,
"learning_rate": 8.98e-05,
"loss": 0.3707286834716797,
"step": 450
},
{
"epoch": 0.012392655743933828,
"grad_norm": 0.20325998961925507,
"learning_rate": 9.180000000000001e-05,
"loss": 0.37896702289581297,
"step": 460
},
{
"epoch": 0.012684935360536042,
"grad_norm": 0.20927369594573975,
"learning_rate": 9.38e-05,
"loss": 0.4257774829864502,
"step": 470
},
{
"epoch": 0.012977214977138253,
"grad_norm": 0.2156561017036438,
"learning_rate": 9.58e-05,
"loss": 0.33562607765197755,
"step": 480
},
{
"epoch": 0.013269494593740466,
"grad_norm": 0.17501753568649292,
"learning_rate": 9.78e-05,
"loss": 0.4147501945495605,
"step": 490
},
{
"epoch": 0.01356177421034268,
"grad_norm": 0.243160679936409,
"learning_rate": 9.98e-05,
"loss": 0.411363410949707,
"step": 500
},
{
"epoch": 0.013854053826944893,
"grad_norm": 0.2272544652223587,
"learning_rate": 0.00010180000000000001,
"loss": 0.36809728145599363,
"step": 510
},
{
"epoch": 0.014146333443547106,
"grad_norm": 0.2265583574771881,
"learning_rate": 0.0001038,
"loss": 0.352154016494751,
"step": 520
},
{
"epoch": 0.014438613060149319,
"grad_norm": 0.21968695521354675,
"learning_rate": 0.00010580000000000001,
"loss": 0.40874533653259276,
"step": 530
},
{
"epoch": 0.01473089267675153,
"grad_norm": 0.20146295428276062,
"learning_rate": 0.00010780000000000002,
"loss": 0.4024982452392578,
"step": 540
},
{
"epoch": 0.015023172293353744,
"grad_norm": 0.22711141407489777,
"learning_rate": 0.00010980000000000001,
"loss": 0.412246561050415,
"step": 550
},
{
"epoch": 0.015315451909955957,
"grad_norm": 0.2056141048669815,
"learning_rate": 0.00011180000000000002,
"loss": 0.4331079006195068,
"step": 560
},
{
"epoch": 0.01560773152655817,
"grad_norm": 0.18964128196239471,
"learning_rate": 0.0001138,
"loss": 0.4245738506317139,
"step": 570
},
{
"epoch": 0.01590001114316038,
"grad_norm": 0.18264859914779663,
"learning_rate": 0.0001158,
"loss": 0.3981050491333008,
"step": 580
},
{
"epoch": 0.016192290759762595,
"grad_norm": 0.198320671916008,
"learning_rate": 0.0001178,
"loss": 0.44843344688415526,
"step": 590
},
{
"epoch": 0.016484570376364808,
"grad_norm": 0.20583713054656982,
"learning_rate": 0.0001198,
"loss": 0.45852208137512207,
"step": 600
},
{
"epoch": 0.01677684999296702,
"grad_norm": 0.18764640390872955,
"learning_rate": 0.0001218,
"loss": 0.39210479259490966,
"step": 610
},
{
"epoch": 0.017069129609569234,
"grad_norm": 0.22744227945804596,
"learning_rate": 0.0001238,
"loss": 0.48337244987487793,
"step": 620
},
{
"epoch": 0.017361409226171447,
"grad_norm": 0.25332197546958923,
"learning_rate": 0.0001258,
"loss": 0.47956342697143556,
"step": 630
},
{
"epoch": 0.01765368884277366,
"grad_norm": 0.18094617128372192,
"learning_rate": 0.00012780000000000002,
"loss": 0.36708922386169435,
"step": 640
},
{
"epoch": 0.017945968459375874,
"grad_norm": 0.2056637406349182,
"learning_rate": 0.0001298,
"loss": 0.45494647026062013,
"step": 650
},
{
"epoch": 0.018238248075978087,
"grad_norm": 0.18681053817272186,
"learning_rate": 0.0001318,
"loss": 0.39863851070404055,
"step": 660
},
{
"epoch": 0.0185305276925803,
"grad_norm": 0.18398210406303406,
"learning_rate": 0.00013380000000000003,
"loss": 0.42660346031188967,
"step": 670
},
{
"epoch": 0.018822807309182513,
"grad_norm": 0.2018873244524002,
"learning_rate": 0.00013580000000000002,
"loss": 0.41592912673950194,
"step": 680
},
{
"epoch": 0.019115086925784727,
"grad_norm": 0.20749719440937042,
"learning_rate": 0.0001378,
"loss": 0.4529616355895996,
"step": 690
},
{
"epoch": 0.01940736654238694,
"grad_norm": 0.17765171825885773,
"learning_rate": 0.0001398,
"loss": 0.4367457389831543,
"step": 700
},
{
"epoch": 0.01969964615898915,
"grad_norm": 0.24419377744197845,
"learning_rate": 0.0001418,
"loss": 0.4324330806732178,
"step": 710
},
{
"epoch": 0.019991925775591363,
"grad_norm": 0.1319461613893509,
"learning_rate": 0.0001438,
"loss": 0.38980741500854493,
"step": 720
},
{
"epoch": 0.020284205392193576,
"grad_norm": 0.18414747714996338,
"learning_rate": 0.0001458,
"loss": 0.45140395164489744,
"step": 730
},
{
"epoch": 0.02057648500879579,
"grad_norm": 0.19065886735916138,
"learning_rate": 0.00014780000000000001,
"loss": 0.34149444103240967,
"step": 740
},
{
"epoch": 0.020868764625398002,
"grad_norm": 0.3047577738761902,
"learning_rate": 0.0001498,
"loss": 0.39748594760894773,
"step": 750
},
{
"epoch": 0.021161044242000215,
"grad_norm": 0.20547902584075928,
"learning_rate": 0.0001518,
"loss": 0.35707693099975585,
"step": 760
},
{
"epoch": 0.02145332385860243,
"grad_norm": 0.1835167109966278,
"learning_rate": 0.0001538,
"loss": 0.3866844177246094,
"step": 770
},
{
"epoch": 0.021745603475204642,
"grad_norm": 0.12297193706035614,
"learning_rate": 0.00015580000000000002,
"loss": 0.3809194803237915,
"step": 780
},
{
"epoch": 0.022037883091806855,
"grad_norm": 0.23474228382110596,
"learning_rate": 0.00015780000000000001,
"loss": 0.40447239875793456,
"step": 790
},
{
"epoch": 0.022330162708409068,
"grad_norm": 0.1950707882642746,
"learning_rate": 0.0001598,
"loss": 0.36057252883911134,
"step": 800
},
{
"epoch": 0.02262244232501128,
"grad_norm": 0.19598263502120972,
"learning_rate": 0.00016180000000000003,
"loss": 0.3532793760299683,
"step": 810
},
{
"epoch": 0.022914721941613495,
"grad_norm": 0.21619845926761627,
"learning_rate": 0.0001638,
"loss": 0.33423144817352296,
"step": 820
},
{
"epoch": 0.023207001558215708,
"grad_norm": 0.1934775710105896,
"learning_rate": 0.0001658,
"loss": 0.4006977558135986,
"step": 830
},
{
"epoch": 0.023499281174817917,
"grad_norm": 0.2345370650291443,
"learning_rate": 0.0001678,
"loss": 0.42209348678588865,
"step": 840
},
{
"epoch": 0.02379156079142013,
"grad_norm": 0.2699654698371887,
"learning_rate": 0.0001698,
"loss": 0.3334143400192261,
"step": 850
},
{
"epoch": 0.024083840408022344,
"grad_norm": 0.1839752197265625,
"learning_rate": 0.0001718,
"loss": 0.42084970474243166,
"step": 860
},
{
"epoch": 0.024376120024624557,
"grad_norm": 0.22491669654846191,
"learning_rate": 0.0001738,
"loss": 0.420766544342041,
"step": 870
},
{
"epoch": 0.02466839964122677,
"grad_norm": 0.18370424211025238,
"learning_rate": 0.00017580000000000002,
"loss": 0.3486522912979126,
"step": 880
},
{
"epoch": 0.024960679257828983,
"grad_norm": 0.23166073858737946,
"learning_rate": 0.0001778,
"loss": 0.42749767303466796,
"step": 890
},
{
"epoch": 0.025252958874431197,
"grad_norm": 0.22481808066368103,
"learning_rate": 0.0001798,
"loss": 0.380033016204834,
"step": 900
},
{
"epoch": 0.02554523849103341,
"grad_norm": 0.24065051972866058,
"learning_rate": 0.00018180000000000003,
"loss": 0.45965576171875,
"step": 910
},
{
"epoch": 0.025837518107635623,
"grad_norm": 0.190377876162529,
"learning_rate": 0.00018380000000000002,
"loss": 0.3290415287017822,
"step": 920
},
{
"epoch": 0.026129797724237836,
"grad_norm": 0.2699977159500122,
"learning_rate": 0.00018580000000000002,
"loss": 0.4352889537811279,
"step": 930
},
{
"epoch": 0.02642207734084005,
"grad_norm": 0.2756347060203552,
"learning_rate": 0.0001878,
"loss": 0.4288458347320557,
"step": 940
},
{
"epoch": 0.026714356957442262,
"grad_norm": 0.23565548658370972,
"learning_rate": 0.0001898,
"loss": 0.29646151065826415,
"step": 950
},
{
"epoch": 0.027006636574044476,
"grad_norm": 0.18665672838687897,
"learning_rate": 0.0001918,
"loss": 0.39093432426452634,
"step": 960
},
{
"epoch": 0.027298916190646685,
"grad_norm": 0.24153801798820496,
"learning_rate": 0.0001938,
"loss": 0.42369751930236815,
"step": 970
},
{
"epoch": 0.0275911958072489,
"grad_norm": 0.18671485781669617,
"learning_rate": 0.00019580000000000002,
"loss": 0.38338029384613037,
"step": 980
},
{
"epoch": 0.027883475423851112,
"grad_norm": 0.2615300118923187,
"learning_rate": 0.0001978,
"loss": 0.36945343017578125,
"step": 990
},
{
"epoch": 0.028175755040453325,
"grad_norm": 0.2992021143436432,
"learning_rate": 0.0001998,
"loss": 0.4174357891082764,
"step": 1000
}
],
"logging_steps": 10,
"max_steps": 40000,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 1000,
"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.06641651400704e+18,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}