| { |
| "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 |
| } |
|
|
