| { | |
| "best_global_step": 3500, | |
| "best_metric": 1.2840659618377686, | |
| "best_model_checkpoint": "/home/hyang45/se_git_repo/final_addtional_exp_mia_trans/save/new/microsoft_CodeGPT-small-java/1/checkpoint-3500", | |
| "epoch": 5.0, | |
| "eval_steps": 500, | |
| "global_step": 4975, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.05027652086475616, | |
| "grad_norm": 0.38967880606651306, | |
| "learning_rate": 3.9200000000000004e-05, | |
| "loss": 1.4062, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.10055304172951232, | |
| "grad_norm": 0.25512877106666565, | |
| "learning_rate": 7.92e-05, | |
| "loss": 0.9927, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.15082956259426847, | |
| "grad_norm": 0.2808470129966736, | |
| "learning_rate": 7.998005947565195e-05, | |
| "loss": 0.8156, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.20110608345902464, | |
| "grad_norm": 0.2150183469057083, | |
| "learning_rate": 7.991862264348792e-05, | |
| "loss": 0.7848, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.2513826043237808, | |
| "grad_norm": 0.2027038335800171, | |
| "learning_rate": 7.981574497732115e-05, | |
| "loss": 0.764, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.30165912518853694, | |
| "grad_norm": 0.23373721539974213, | |
| "learning_rate": 7.967153327783776e-05, | |
| "loss": 0.7302, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.35193564605329314, | |
| "grad_norm": 0.18324315547943115, | |
| "learning_rate": 7.948613725594216e-05, | |
| "loss": 0.7287, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.4022121669180493, | |
| "grad_norm": 0.21186716854572296, | |
| "learning_rate": 7.925974937733732e-05, | |
| "loss": 0.7011, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.45248868778280543, | |
| "grad_norm": 0.3749410808086395, | |
| "learning_rate": 7.899260466271975e-05, | |
| "loss": 0.7029, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.5027652086475616, | |
| "grad_norm": 0.1961645632982254, | |
| "learning_rate": 7.868498044379681e-05, | |
| "loss": 0.6967, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.5027652086475616, | |
| "eval_accuracy": 0.7520552872875925, | |
| "eval_loss": 1.3162076473236084, | |
| "eval_runtime": 85.0292, | |
| "eval_samples_per_second": 57.91, | |
| "eval_steps_per_second": 2.423, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.5530417295123178, | |
| "grad_norm": 0.17420420050621033, | |
| "learning_rate": 7.833719607537954e-05, | |
| "loss": 0.6851, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.6033182503770739, | |
| "grad_norm": 0.2493220567703247, | |
| "learning_rate": 7.794961260384997e-05, | |
| "loss": 0.671, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.6535947712418301, | |
| "grad_norm": 0.20172835886478424, | |
| "learning_rate": 7.752263239234692e-05, | |
| "loss": 0.662, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.7038712921065863, | |
| "grad_norm": 0.25561460852622986, | |
| "learning_rate": 7.705669870305975e-05, | |
| "loss": 0.6763, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.7541478129713424, | |
| "grad_norm": 0.19745612144470215, | |
| "learning_rate": 7.655229523706317e-05, | |
| "loss": 0.6545, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.8044243338360986, | |
| "grad_norm": 0.1649468094110489, | |
| "learning_rate": 7.600994563217132e-05, | |
| "loss": 0.6463, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.8547008547008547, | |
| "grad_norm": 0.20483672618865967, | |
| "learning_rate": 7.543021291933214e-05, | |
| "loss": 0.6625, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.9049773755656109, | |
| "grad_norm": 0.1755620539188385, | |
| "learning_rate": 7.481369893812647e-05, | |
| "loss": 0.6629, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.9552538964303671, | |
| "grad_norm": 0.1850636750459671, | |
| "learning_rate": 7.416104371197853e-05, | |
| "loss": 0.6329, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 1.0050276520864756, | |
| "grad_norm": 0.17517416179180145, | |
| "learning_rate": 7.347292478372667e-05, | |
| "loss": 0.6481, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.0050276520864756, | |
| "eval_accuracy": 0.7550630780433458, | |
| "eval_loss": 1.3002119064331055, | |
| "eval_runtime": 91.3368, | |
| "eval_samples_per_second": 53.91, | |
| "eval_steps_per_second": 2.255, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.0553041729512318, | |
| "grad_norm": 0.14569388329982758, | |
| "learning_rate": 7.275005651224378e-05, | |
| "loss": 0.6276, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 1.105580693815988, | |
| "grad_norm": 0.1603846698999405, | |
| "learning_rate": 7.199318933083797e-05, | |
| "loss": 0.6346, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.155857214680744, | |
| "grad_norm": 0.20617897808551788, | |
| "learning_rate": 7.120310896820298e-05, | |
| "loss": 0.6191, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 1.2061337355455002, | |
| "grad_norm": 0.2762608230113983, | |
| "learning_rate": 7.038063563272751e-05, | |
| "loss": 0.63, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.2564102564102564, | |
| "grad_norm": 0.13912968337535858, | |
| "learning_rate": 6.952662316100984e-05, | |
| "loss": 0.6265, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 1.3066867772750126, | |
| "grad_norm": 0.27741900086402893, | |
| "learning_rate": 6.864195813146212e-05, | |
| "loss": 0.6204, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.3569632981397688, | |
| "grad_norm": 0.15955646336078644, | |
| "learning_rate": 6.772755894392405e-05, | |
| "loss": 0.6077, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 1.4072398190045248, | |
| "grad_norm": 0.15095743536949158, | |
| "learning_rate": 6.678437486624186e-05, | |
| "loss": 0.6361, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.457516339869281, | |
| "grad_norm": 0.14820362627506256, | |
| "learning_rate": 6.58133850488023e-05, | |
| "loss": 0.6286, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 1.5077928607340372, | |
| "grad_norm": 0.16185027360916138, | |
| "learning_rate": 6.481559750804423e-05, | |
| "loss": 0.6091, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.5077928607340372, | |
| "eval_accuracy": 0.7550170211853606, | |
| "eval_loss": 1.2964880466461182, | |
| "eval_runtime": 85.294, | |
| "eval_samples_per_second": 57.73, | |
| "eval_steps_per_second": 2.415, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.5580693815987934, | |
| "grad_norm": 0.16229036450386047, | |
| "learning_rate": 6.379204808000379e-05, | |
| "loss": 0.6024, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 1.6083459024635496, | |
| "grad_norm": 0.14892268180847168, | |
| "learning_rate": 6.274379934497889e-05, | |
| "loss": 0.605, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 1.6586224233283056, | |
| "grad_norm": 0.15874284505844116, | |
| "learning_rate": 6.167193952442972e-05, | |
| "loss": 0.5853, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 1.708898944193062, | |
| "grad_norm": 0.16238285601139069, | |
| "learning_rate": 6.057758135126026e-05, | |
| "loss": 0.6054, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 1.759175465057818, | |
| "grad_norm": 0.16057848930358887, | |
| "learning_rate": 5.946186091465375e-05, | |
| "loss": 0.6252, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 1.8094519859225742, | |
| "grad_norm": 0.2041744887828827, | |
| "learning_rate": 5.8325936480661065e-05, | |
| "loss": 0.5942, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 1.8597285067873304, | |
| "grad_norm": 0.23550090193748474, | |
| "learning_rate": 5.717098728976677e-05, | |
| "loss": 0.6056, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 1.9100050276520864, | |
| "grad_norm": 0.2757298946380615, | |
| "learning_rate": 5.599821233268077e-05, | |
| "loss": 0.6023, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 1.9602815485168428, | |
| "grad_norm": 0.1491357684135437, | |
| "learning_rate": 5.4808829105626623e-05, | |
| "loss": 0.6097, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 2.010055304172951, | |
| "grad_norm": 0.21005308628082275, | |
| "learning_rate": 5.360407234641879e-05, | |
| "loss": 0.6091, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.010055304172951, | |
| "eval_accuracy": 0.7556497074198393, | |
| "eval_loss": 1.289443016052246, | |
| "eval_runtime": 85.2093, | |
| "eval_samples_per_second": 57.787, | |
| "eval_steps_per_second": 2.418, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.060331825037707, | |
| "grad_norm": 0.15782801806926727, | |
| "learning_rate": 5.238519275264064e-05, | |
| "loss": 0.5902, | |
| "step": 2050 | |
| }, | |
| { | |
| "epoch": 2.1106083459024636, | |
| "grad_norm": 0.1624271422624588, | |
| "learning_rate": 5.1153455683254215e-05, | |
| "loss": 0.5942, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.1608848667672196, | |
| "grad_norm": 0.16978579759597778, | |
| "learning_rate": 4.9910139844989574e-05, | |
| "loss": 0.5845, | |
| "step": 2150 | |
| }, | |
| { | |
| "epoch": 2.211161387631976, | |
| "grad_norm": 0.14723461866378784, | |
| "learning_rate": 4.865653596487719e-05, | |
| "loss": 0.5954, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.261437908496732, | |
| "grad_norm": 0.16163045167922974, | |
| "learning_rate": 4.739394545030192e-05, | |
| "loss": 0.5821, | |
| "step": 2250 | |
| }, | |
| { | |
| "epoch": 2.311714429361488, | |
| "grad_norm": 0.22584794461727142, | |
| "learning_rate": 4.6123679037969156e-05, | |
| "loss": 0.5811, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.3619909502262444, | |
| "grad_norm": 0.16639545559883118, | |
| "learning_rate": 4.484705543318603e-05, | |
| "loss": 0.5923, | |
| "step": 2350 | |
| }, | |
| { | |
| "epoch": 2.4122674710910004, | |
| "grad_norm": 0.14507165551185608, | |
| "learning_rate": 4.356539994087004e-05, | |
| "loss": 0.5758, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 2.462543991955757, | |
| "grad_norm": 0.17524458467960358, | |
| "learning_rate": 4.228004308970651e-05, | |
| "loss": 0.5793, | |
| "step": 2450 | |
| }, | |
| { | |
| "epoch": 2.5128205128205128, | |
| "grad_norm": 0.20955400168895721, | |
| "learning_rate": 4.099231925088303e-05, | |
| "loss": 0.5805, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.5128205128205128, | |
| "eval_accuracy": 0.7555808206538009, | |
| "eval_loss": 1.2918362617492676, | |
| "eval_runtime": 84.9046, | |
| "eval_samples_per_second": 57.994, | |
| "eval_steps_per_second": 2.426, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.5630970336852688, | |
| "grad_norm": 0.15553168952465057, | |
| "learning_rate": 3.970356525283485e-05, | |
| "loss": 0.5902, | |
| "step": 2550 | |
| }, | |
| { | |
| "epoch": 2.613373554550025, | |
| "grad_norm": 0.24212880432605743, | |
| "learning_rate": 3.841511899343937e-05, | |
| "loss": 0.5896, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 2.663650075414781, | |
| "grad_norm": 0.14325997233390808, | |
| "learning_rate": 3.712831805110037e-05, | |
| "loss": 0.5821, | |
| "step": 2650 | |
| }, | |
| { | |
| "epoch": 2.7139265962795376, | |
| "grad_norm": 0.16522951424121857, | |
| "learning_rate": 3.584449829616395e-05, | |
| "loss": 0.5842, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 2.7642031171442936, | |
| "grad_norm": 0.12420392781496048, | |
| "learning_rate": 3.456499250410759e-05, | |
| "loss": 0.5844, | |
| "step": 2750 | |
| }, | |
| { | |
| "epoch": 2.8144796380090495, | |
| "grad_norm": 0.17248398065567017, | |
| "learning_rate": 3.3291128971942105e-05, | |
| "loss": 0.587, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 2.864756158873806, | |
| "grad_norm": 0.14520220458507538, | |
| "learning_rate": 3.2024230139262936e-05, | |
| "loss": 0.5837, | |
| "step": 2850 | |
| }, | |
| { | |
| "epoch": 2.915032679738562, | |
| "grad_norm": 0.14959931373596191, | |
| "learning_rate": 3.076561121538211e-05, | |
| "loss": 0.5827, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 2.9653092006033184, | |
| "grad_norm": 0.2161024659872055, | |
| "learning_rate": 2.9516578813966154e-05, | |
| "loss": 0.5717, | |
| "step": 2950 | |
| }, | |
| { | |
| "epoch": 3.015082956259427, | |
| "grad_norm": 0.14117109775543213, | |
| "learning_rate": 2.827842959659771e-05, | |
| "loss": 0.557, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.015082956259427, | |
| "eval_accuracy": 0.7559059979528521, | |
| "eval_loss": 1.2862486839294434, | |
| "eval_runtime": 85.3029, | |
| "eval_samples_per_second": 57.724, | |
| "eval_steps_per_second": 2.415, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.065359477124183, | |
| "grad_norm": 0.180222749710083, | |
| "learning_rate": 2.7052448926668305e-05, | |
| "loss": 0.5686, | |
| "step": 3050 | |
| }, | |
| { | |
| "epoch": 3.115635997988939, | |
| "grad_norm": 0.2005385011434555, | |
| "learning_rate": 2.5839909535000498e-05, | |
| "loss": 0.5682, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 3.165912518853695, | |
| "grad_norm": 0.18873438239097595, | |
| "learning_rate": 2.4642070198584236e-05, | |
| "loss": 0.5816, | |
| "step": 3150 | |
| }, | |
| { | |
| "epoch": 3.2161890397184516, | |
| "grad_norm": 0.18745166063308716, | |
| "learning_rate": 2.3460174433798927e-05, | |
| "loss": 0.5739, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 3.2664655605832076, | |
| "grad_norm": 0.17067401111125946, | |
| "learning_rate": 2.2295449205478362e-05, | |
| "loss": 0.5864, | |
| "step": 3250 | |
| }, | |
| { | |
| "epoch": 3.3167420814479636, | |
| "grad_norm": 0.16623741388320923, | |
| "learning_rate": 2.114910365315819e-05, | |
| "loss": 0.5795, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 3.36701860231272, | |
| "grad_norm": 0.15764974057674408, | |
| "learning_rate": 2.0022327835828362e-05, | |
| "loss": 0.5704, | |
| "step": 3350 | |
| }, | |
| { | |
| "epoch": 3.417295123177476, | |
| "grad_norm": 0.14753426611423492, | |
| "learning_rate": 1.891629149649378e-05, | |
| "loss": 0.5633, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 3.4675716440422324, | |
| "grad_norm": 0.3436726927757263, | |
| "learning_rate": 1.7832142847825763e-05, | |
| "loss": 0.5753, | |
| "step": 3450 | |
| }, | |
| { | |
| "epoch": 3.5178481649069884, | |
| "grad_norm": 0.1702040284872055, | |
| "learning_rate": 1.677100738016461e-05, | |
| "loss": 0.5513, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 3.5178481649069884, | |
| "eval_accuracy": 0.7564910391618287, | |
| "eval_loss": 1.2840659618377686, | |
| "eval_runtime": 85.0647, | |
| "eval_samples_per_second": 57.885, | |
| "eval_steps_per_second": 2.422, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 3.5681246857717444, | |
| "grad_norm": 0.18295463919639587, | |
| "learning_rate": 1.573398669311116e-05, | |
| "loss": 0.5823, | |
| "step": 3550 | |
| }, | |
| { | |
| "epoch": 3.6184012066365008, | |
| "grad_norm": 0.17615950107574463, | |
| "learning_rate": 1.472215735192e-05, | |
| "loss": 0.5645, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 3.6686777275012568, | |
| "grad_norm": 0.2151349037885666, | |
| "learning_rate": 1.373656976988175e-05, | |
| "loss": 0.5725, | |
| "step": 3650 | |
| }, | |
| { | |
| "epoch": 3.718954248366013, | |
| "grad_norm": 0.13829338550567627, | |
| "learning_rate": 1.2778247117854439e-05, | |
| "loss": 0.5576, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 3.769230769230769, | |
| "grad_norm": 0.16056522727012634, | |
| "learning_rate": 1.1848184262076306e-05, | |
| "loss": 0.5768, | |
| "step": 3750 | |
| }, | |
| { | |
| "epoch": 3.819507290095525, | |
| "grad_norm": 0.18423378467559814, | |
| "learning_rate": 1.0947346731362357e-05, | |
| "loss": 0.5681, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 3.8697838109602816, | |
| "grad_norm": 0.15498067438602448, | |
| "learning_rate": 1.0076669714757261e-05, | |
| "loss": 0.5507, | |
| "step": 3850 | |
| }, | |
| { | |
| "epoch": 3.9200603318250375, | |
| "grad_norm": 0.15752053260803223, | |
| "learning_rate": 9.237057090684773e-06, | |
| "loss": 0.5641, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 3.970336852689794, | |
| "grad_norm": 0.14952270686626434, | |
| "learning_rate": 8.42938048860198e-06, | |
| "loss": 0.5704, | |
| "step": 3950 | |
| }, | |
| { | |
| "epoch": 4.020110608345902, | |
| "grad_norm": 0.13633842766284943, | |
| "learning_rate": 7.654478384131975e-06, | |
| "loss": 0.5468, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 4.020110608345902, | |
| "eval_accuracy": 0.7562369323591513, | |
| "eval_loss": 1.2848825454711914, | |
| "eval_runtime": 85.0878, | |
| "eval_samples_per_second": 57.87, | |
| "eval_steps_per_second": 2.421, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 4.070387129210658, | |
| "grad_norm": 0.1506165862083435, | |
| "learning_rate": 6.913155228614838e-06, | |
| "loss": 0.5674, | |
| "step": 4050 | |
| }, | |
| { | |
| "epoch": 4.120663650075414, | |
| "grad_norm": 0.15265598893165588, | |
| "learning_rate": 6.206180613980266e-06, | |
| "loss": 0.5625, | |
| "step": 4100 | |
| }, | |
| { | |
| "epoch": 4.170940170940171, | |
| "grad_norm": 0.13488344848155975, | |
| "learning_rate": 5.534288473808831e-06, | |
| "loss": 0.5624, | |
| "step": 4150 | |
| }, | |
| { | |
| "epoch": 4.221216691804927, | |
| "grad_norm": 0.13856124877929688, | |
| "learning_rate": 4.898176321411478e-06, | |
| "loss": 0.5628, | |
| "step": 4200 | |
| }, | |
| { | |
| "epoch": 4.271493212669683, | |
| "grad_norm": 0.18089675903320312, | |
| "learning_rate": 4.2985045257180415e-06, | |
| "loss": 0.5731, | |
| "step": 4250 | |
| }, | |
| { | |
| "epoch": 4.321769733534439, | |
| "grad_norm": 0.14551880955696106, | |
| "learning_rate": 3.735895625726493e-06, | |
| "loss": 0.5562, | |
| "step": 4300 | |
| }, | |
| { | |
| "epoch": 4.372046254399196, | |
| "grad_norm": 0.13140977919101715, | |
| "learning_rate": 3.210933684224804e-06, | |
| "loss": 0.5729, | |
| "step": 4350 | |
| }, | |
| { | |
| "epoch": 4.422322775263952, | |
| "grad_norm": 0.16188330948352814, | |
| "learning_rate": 2.7241636814561514e-06, | |
| "loss": 0.5633, | |
| "step": 4400 | |
| }, | |
| { | |
| "epoch": 4.472599296128708, | |
| "grad_norm": 0.14776812493801117, | |
| "learning_rate": 2.2760909493570127e-06, | |
| "loss": 0.5677, | |
| "step": 4450 | |
| }, | |
| { | |
| "epoch": 4.522875816993464, | |
| "grad_norm": 0.15746714174747467, | |
| "learning_rate": 1.8671806469554976e-06, | |
| "loss": 0.5513, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 4.522875816993464, | |
| "eval_accuracy": 0.7559050053481541, | |
| "eval_loss": 1.2894773483276367, | |
| "eval_runtime": 85.2306, | |
| "eval_samples_per_second": 57.773, | |
| "eval_steps_per_second": 2.417, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 4.57315233785822, | |
| "grad_norm": 0.1549079418182373, | |
| "learning_rate": 1.497857277474455e-06, | |
| "loss": 0.5727, | |
| "step": 4550 | |
| }, | |
| { | |
| "epoch": 4.623428858722976, | |
| "grad_norm": 0.1668921858072281, | |
| "learning_rate": 1.1685042476407404e-06, | |
| "loss": 0.5641, | |
| "step": 4600 | |
| }, | |
| { | |
| "epoch": 4.673705379587733, | |
| "grad_norm": 0.134994775056839, | |
| "learning_rate": 8.794634696580551e-07, | |
| "loss": 0.5597, | |
| "step": 4650 | |
| }, | |
| { | |
| "epoch": 4.723981900452489, | |
| "grad_norm": 0.1448977142572403, | |
| "learning_rate": 6.310350062567194e-07, | |
| "loss": 0.5672, | |
| "step": 4700 | |
| }, | |
| { | |
| "epoch": 4.774258421317245, | |
| "grad_norm": 0.16990165412425995, | |
| "learning_rate": 4.2347675918863463e-07, | |
| "loss": 0.5669, | |
| "step": 4750 | |
| }, | |
| { | |
| "epoch": 4.824534942182001, | |
| "grad_norm": 0.16192808747291565, | |
| "learning_rate": 2.5700420149102233e-07, | |
| "loss": 0.5778, | |
| "step": 4800 | |
| }, | |
| { | |
| "epoch": 4.874811463046758, | |
| "grad_norm": 0.154259592294693, | |
| "learning_rate": 1.3179015379677585e-07, | |
| "loss": 0.5774, | |
| "step": 4850 | |
| }, | |
| { | |
| "epoch": 4.925087983911514, | |
| "grad_norm": 0.16326895356178284, | |
| "learning_rate": 4.796460492368926e-08, | |
| "loss": 0.5586, | |
| "step": 4900 | |
| }, | |
| { | |
| "epoch": 4.97536450477627, | |
| "grad_norm": 0.18583731353282928, | |
| "learning_rate": 5.614576928785198e-09, | |
| "loss": 0.5682, | |
| "step": 4950 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "step": 4975, | |
| "total_flos": 8.315357626368e+16, | |
| "train_loss": 0.615904873986939, | |
| "train_runtime": 4775.4033, | |
| "train_samples_per_second": 33.321, | |
| "train_steps_per_second": 1.042 | |
| } | |
| ], | |
| "logging_steps": 50, | |
| "max_steps": 4975, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 8.315357626368e+16, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |