koja_mbartLarge_55p_run1 / trainer_state.json
yesj1234's picture
Upload folder using huggingface_hub
6ab071f
raw
history blame contribute delete
No virus
18.5 kB
{
"best_metric": 0.9008079171180725,
"best_model_checkpoint": "./koja_mbartLarge_55p_run1/checkpoint-32000",
"epoch": 3.8648509918777743,
"eval_steps": 8000,
"global_step": 64000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.03,
"learning_rate": 4.9849024699559153e-05,
"loss": 1.5854,
"step": 500
},
{
"epoch": 0.06,
"learning_rate": 4.9698049399118304e-05,
"loss": 1.3951,
"step": 1000
},
{
"epoch": 0.09,
"learning_rate": 4.954707409867746e-05,
"loss": 1.2938,
"step": 1500
},
{
"epoch": 0.12,
"learning_rate": 4.939609879823661e-05,
"loss": 1.2482,
"step": 2000
},
{
"epoch": 0.15,
"learning_rate": 4.9245123497795764e-05,
"loss": 1.2122,
"step": 2500
},
{
"epoch": 0.18,
"learning_rate": 4.9094148197354915e-05,
"loss": 1.178,
"step": 3000
},
{
"epoch": 0.21,
"learning_rate": 4.8943172896914066e-05,
"loss": 1.1483,
"step": 3500
},
{
"epoch": 0.24,
"learning_rate": 4.8792197596473224e-05,
"loss": 1.1408,
"step": 4000
},
{
"epoch": 0.27,
"learning_rate": 4.8641222296032375e-05,
"loss": 1.1053,
"step": 4500
},
{
"epoch": 0.3,
"learning_rate": 4.8490246995591526e-05,
"loss": 1.0876,
"step": 5000
},
{
"epoch": 0.33,
"learning_rate": 4.833927169515067e-05,
"loss": 1.0745,
"step": 5500
},
{
"epoch": 0.36,
"learning_rate": 4.818829639470983e-05,
"loss": 1.0707,
"step": 6000
},
{
"epoch": 0.39,
"learning_rate": 4.803732109426898e-05,
"loss": 1.0664,
"step": 6500
},
{
"epoch": 0.42,
"learning_rate": 4.788634579382813e-05,
"loss": 1.0426,
"step": 7000
},
{
"epoch": 0.45,
"learning_rate": 4.773537049338728e-05,
"loss": 1.0357,
"step": 7500
},
{
"epoch": 0.48,
"learning_rate": 4.758439519294643e-05,
"loss": 1.0254,
"step": 8000
},
{
"epoch": 0.48,
"eval_bleu": 7.1252,
"eval_gen_len": 18.5403,
"eval_loss": 1.0065276622772217,
"eval_runtime": 2330.5491,
"eval_samples_per_second": 14.21,
"eval_steps_per_second": 0.888,
"step": 8000
},
{
"epoch": 0.51,
"learning_rate": 4.743341989250559e-05,
"loss": 1.0195,
"step": 8500
},
{
"epoch": 0.54,
"learning_rate": 4.728244459206474e-05,
"loss": 1.0094,
"step": 9000
},
{
"epoch": 0.57,
"learning_rate": 4.713146929162389e-05,
"loss": 0.9984,
"step": 9500
},
{
"epoch": 0.6,
"learning_rate": 4.698049399118304e-05,
"loss": 0.9693,
"step": 10000
},
{
"epoch": 0.63,
"learning_rate": 4.6829518690742194e-05,
"loss": 0.9945,
"step": 10500
},
{
"epoch": 0.66,
"learning_rate": 4.667854339030135e-05,
"loss": 0.989,
"step": 11000
},
{
"epoch": 0.69,
"learning_rate": 4.65275680898605e-05,
"loss": 0.9907,
"step": 11500
},
{
"epoch": 0.72,
"learning_rate": 4.6376592789419654e-05,
"loss": 0.97,
"step": 12000
},
{
"epoch": 0.75,
"learning_rate": 4.6225617488978805e-05,
"loss": 0.9642,
"step": 12500
},
{
"epoch": 0.79,
"learning_rate": 4.6074642188537956e-05,
"loss": 0.9615,
"step": 13000
},
{
"epoch": 0.82,
"learning_rate": 4.5923666888097114e-05,
"loss": 0.9485,
"step": 13500
},
{
"epoch": 0.85,
"learning_rate": 4.5772691587656265e-05,
"loss": 0.9492,
"step": 14000
},
{
"epoch": 0.88,
"learning_rate": 4.5621716287215416e-05,
"loss": 0.9568,
"step": 14500
},
{
"epoch": 0.91,
"learning_rate": 4.547074098677457e-05,
"loss": 0.9507,
"step": 15000
},
{
"epoch": 0.94,
"learning_rate": 4.531976568633372e-05,
"loss": 0.9437,
"step": 15500
},
{
"epoch": 0.97,
"learning_rate": 4.516879038589287e-05,
"loss": 0.9374,
"step": 16000
},
{
"epoch": 0.97,
"eval_bleu": 8.2047,
"eval_gen_len": 18.2091,
"eval_loss": 0.9199957847595215,
"eval_runtime": 2175.6016,
"eval_samples_per_second": 15.222,
"eval_steps_per_second": 0.951,
"step": 16000
},
{
"epoch": 1.0,
"learning_rate": 4.501781508545202e-05,
"loss": 0.9266,
"step": 16500
},
{
"epoch": 1.03,
"learning_rate": 4.486683978501117e-05,
"loss": 0.8505,
"step": 17000
},
{
"epoch": 1.06,
"learning_rate": 4.471586448457032e-05,
"loss": 0.8316,
"step": 17500
},
{
"epoch": 1.09,
"learning_rate": 4.456488918412948e-05,
"loss": 0.8079,
"step": 18000
},
{
"epoch": 1.12,
"learning_rate": 4.441391388368863e-05,
"loss": 0.7993,
"step": 18500
},
{
"epoch": 1.15,
"learning_rate": 4.426293858324778e-05,
"loss": 0.7984,
"step": 19000
},
{
"epoch": 1.18,
"learning_rate": 4.411196328280693e-05,
"loss": 0.789,
"step": 19500
},
{
"epoch": 1.21,
"learning_rate": 4.3960987982366084e-05,
"loss": 0.7766,
"step": 20000
},
{
"epoch": 1.24,
"learning_rate": 4.381001268192524e-05,
"loss": 0.7798,
"step": 20500
},
{
"epoch": 1.27,
"learning_rate": 4.365903738148439e-05,
"loss": 0.7717,
"step": 21000
},
{
"epoch": 1.3,
"learning_rate": 4.3508062081043544e-05,
"loss": 0.7581,
"step": 21500
},
{
"epoch": 1.33,
"learning_rate": 4.3357086780602695e-05,
"loss": 0.7531,
"step": 22000
},
{
"epoch": 1.36,
"learning_rate": 4.3206111480161846e-05,
"loss": 0.7578,
"step": 22500
},
{
"epoch": 1.39,
"learning_rate": 4.3055136179721003e-05,
"loss": 0.7528,
"step": 23000
},
{
"epoch": 1.42,
"learning_rate": 4.2904160879280154e-05,
"loss": 0.7479,
"step": 23500
},
{
"epoch": 1.45,
"learning_rate": 4.2753185578839306e-05,
"loss": 0.7429,
"step": 24000
},
{
"epoch": 1.45,
"eval_bleu": 8.4622,
"eval_gen_len": 17.9606,
"eval_loss": 0.9085537791252136,
"eval_runtime": 2144.5036,
"eval_samples_per_second": 15.443,
"eval_steps_per_second": 0.965,
"step": 24000
},
{
"epoch": 1.48,
"learning_rate": 4.2602210278398457e-05,
"loss": 0.7292,
"step": 24500
},
{
"epoch": 1.51,
"learning_rate": 4.245123497795761e-05,
"loss": 0.7363,
"step": 25000
},
{
"epoch": 1.54,
"learning_rate": 4.2300259677516765e-05,
"loss": 0.7261,
"step": 25500
},
{
"epoch": 1.57,
"learning_rate": 4.2149284377075916e-05,
"loss": 0.7214,
"step": 26000
},
{
"epoch": 1.6,
"learning_rate": 4.199830907663506e-05,
"loss": 0.7077,
"step": 26500
},
{
"epoch": 1.63,
"learning_rate": 4.184733377619421e-05,
"loss": 0.724,
"step": 27000
},
{
"epoch": 1.66,
"learning_rate": 4.169635847575337e-05,
"loss": 0.7286,
"step": 27500
},
{
"epoch": 1.69,
"learning_rate": 4.154538317531252e-05,
"loss": 0.7224,
"step": 28000
},
{
"epoch": 1.72,
"learning_rate": 4.139440787487167e-05,
"loss": 0.7194,
"step": 28500
},
{
"epoch": 1.75,
"learning_rate": 4.124343257443082e-05,
"loss": 0.7095,
"step": 29000
},
{
"epoch": 1.78,
"learning_rate": 4.1092457273989973e-05,
"loss": 0.7065,
"step": 29500
},
{
"epoch": 1.81,
"learning_rate": 4.094148197354913e-05,
"loss": 0.7012,
"step": 30000
},
{
"epoch": 1.84,
"learning_rate": 4.079050667310828e-05,
"loss": 0.7045,
"step": 30500
},
{
"epoch": 1.87,
"learning_rate": 4.063953137266743e-05,
"loss": 0.7099,
"step": 31000
},
{
"epoch": 1.9,
"learning_rate": 4.0488556072226584e-05,
"loss": 0.7084,
"step": 31500
},
{
"epoch": 1.93,
"learning_rate": 4.0337580771785735e-05,
"loss": 0.7039,
"step": 32000
},
{
"epoch": 1.93,
"eval_bleu": 8.9198,
"eval_gen_len": 17.8841,
"eval_loss": 0.9008079171180725,
"eval_runtime": 2125.9264,
"eval_samples_per_second": 15.578,
"eval_steps_per_second": 0.974,
"step": 32000
},
{
"epoch": 1.96,
"learning_rate": 4.018660547134489e-05,
"loss": 0.7029,
"step": 32500
},
{
"epoch": 1.99,
"learning_rate": 4.0035630170904044e-05,
"loss": 0.6983,
"step": 33000
},
{
"epoch": 2.02,
"learning_rate": 3.9884654870463195e-05,
"loss": 0.6503,
"step": 33500
},
{
"epoch": 2.05,
"learning_rate": 3.9733679570022346e-05,
"loss": 0.6191,
"step": 34000
},
{
"epoch": 2.08,
"learning_rate": 3.95827042695815e-05,
"loss": 0.605,
"step": 34500
},
{
"epoch": 2.11,
"learning_rate": 3.9431728969140655e-05,
"loss": 0.6035,
"step": 35000
},
{
"epoch": 2.14,
"learning_rate": 3.9280753668699806e-05,
"loss": 0.6008,
"step": 35500
},
{
"epoch": 2.17,
"learning_rate": 3.912977836825896e-05,
"loss": 0.5947,
"step": 36000
},
{
"epoch": 2.2,
"learning_rate": 3.897880306781811e-05,
"loss": 0.5812,
"step": 36500
},
{
"epoch": 2.23,
"learning_rate": 3.882782776737726e-05,
"loss": 0.587,
"step": 37000
},
{
"epoch": 2.26,
"learning_rate": 3.867685246693641e-05,
"loss": 0.5857,
"step": 37500
},
{
"epoch": 2.29,
"learning_rate": 3.852587716649556e-05,
"loss": 0.5659,
"step": 38000
},
{
"epoch": 2.32,
"learning_rate": 3.837490186605471e-05,
"loss": 0.5702,
"step": 38500
},
{
"epoch": 2.36,
"learning_rate": 3.822392656561386e-05,
"loss": 0.5651,
"step": 39000
},
{
"epoch": 2.39,
"learning_rate": 3.807295126517302e-05,
"loss": 0.5691,
"step": 39500
},
{
"epoch": 2.42,
"learning_rate": 3.792197596473217e-05,
"loss": 0.5616,
"step": 40000
},
{
"epoch": 2.42,
"eval_bleu": 8.7938,
"eval_gen_len": 17.78,
"eval_loss": 0.9209699630737305,
"eval_runtime": 2116.9975,
"eval_samples_per_second": 15.644,
"eval_steps_per_second": 0.978,
"step": 40000
},
{
"epoch": 2.45,
"learning_rate": 3.777100066429132e-05,
"loss": 0.5552,
"step": 40500
},
{
"epoch": 2.48,
"learning_rate": 3.7620025363850474e-05,
"loss": 0.5461,
"step": 41000
},
{
"epoch": 2.51,
"learning_rate": 3.7469050063409625e-05,
"loss": 0.5546,
"step": 41500
},
{
"epoch": 2.54,
"learning_rate": 3.731807476296878e-05,
"loss": 0.5383,
"step": 42000
},
{
"epoch": 2.57,
"learning_rate": 3.7167099462527934e-05,
"loss": 0.5452,
"step": 42500
},
{
"epoch": 2.6,
"learning_rate": 3.7016124162087085e-05,
"loss": 0.5318,
"step": 43000
},
{
"epoch": 2.63,
"learning_rate": 3.6865148861646236e-05,
"loss": 0.5407,
"step": 43500
},
{
"epoch": 2.66,
"learning_rate": 3.671417356120539e-05,
"loss": 0.549,
"step": 44000
},
{
"epoch": 2.69,
"learning_rate": 3.6563198260764545e-05,
"loss": 0.5384,
"step": 44500
},
{
"epoch": 2.72,
"learning_rate": 3.6412222960323696e-05,
"loss": 0.5441,
"step": 45000
},
{
"epoch": 2.75,
"learning_rate": 3.626124765988285e-05,
"loss": 0.5317,
"step": 45500
},
{
"epoch": 2.78,
"learning_rate": 3.6110272359442e-05,
"loss": 0.5352,
"step": 46000
},
{
"epoch": 2.81,
"learning_rate": 3.595929705900115e-05,
"loss": 0.5248,
"step": 46500
},
{
"epoch": 2.84,
"learning_rate": 3.5808321758560307e-05,
"loss": 0.5309,
"step": 47000
},
{
"epoch": 2.87,
"learning_rate": 3.565734645811946e-05,
"loss": 0.5296,
"step": 47500
},
{
"epoch": 2.9,
"learning_rate": 3.55063711576786e-05,
"loss": 0.5355,
"step": 48000
},
{
"epoch": 2.9,
"eval_bleu": 8.6506,
"eval_gen_len": 17.9146,
"eval_loss": 0.9399827122688293,
"eval_runtime": 2117.673,
"eval_samples_per_second": 15.639,
"eval_steps_per_second": 0.977,
"step": 48000
},
{
"epoch": 2.93,
"learning_rate": 3.535539585723775e-05,
"loss": 0.5262,
"step": 48500
},
{
"epoch": 2.96,
"learning_rate": 3.520442055679691e-05,
"loss": 0.5247,
"step": 49000
},
{
"epoch": 2.99,
"learning_rate": 3.505344525635606e-05,
"loss": 0.5315,
"step": 49500
},
{
"epoch": 3.02,
"learning_rate": 3.490246995591521e-05,
"loss": 0.4932,
"step": 50000
},
{
"epoch": 3.05,
"learning_rate": 3.4751494655474364e-05,
"loss": 0.4595,
"step": 50500
},
{
"epoch": 3.08,
"learning_rate": 3.4600519355033515e-05,
"loss": 0.45,
"step": 51000
},
{
"epoch": 3.11,
"learning_rate": 3.444954405459267e-05,
"loss": 0.4477,
"step": 51500
},
{
"epoch": 3.14,
"learning_rate": 3.4298568754151824e-05,
"loss": 0.4423,
"step": 52000
},
{
"epoch": 3.17,
"learning_rate": 3.4147593453710975e-05,
"loss": 0.435,
"step": 52500
},
{
"epoch": 3.2,
"learning_rate": 3.3996618153270126e-05,
"loss": 0.4345,
"step": 53000
},
{
"epoch": 3.23,
"learning_rate": 3.3845642852829277e-05,
"loss": 0.4337,
"step": 53500
},
{
"epoch": 3.26,
"learning_rate": 3.3694667552388434e-05,
"loss": 0.4295,
"step": 54000
},
{
"epoch": 3.29,
"learning_rate": 3.3543692251947585e-05,
"loss": 0.4143,
"step": 54500
},
{
"epoch": 3.32,
"learning_rate": 3.3392716951506736e-05,
"loss": 0.4203,
"step": 55000
},
{
"epoch": 3.35,
"learning_rate": 3.324174165106589e-05,
"loss": 0.4164,
"step": 55500
},
{
"epoch": 3.38,
"learning_rate": 3.309076635062504e-05,
"loss": 0.4194,
"step": 56000
},
{
"epoch": 3.38,
"eval_bleu": 8.4235,
"eval_gen_len": 17.6759,
"eval_loss": 1.0053656101226807,
"eval_runtime": 2100.231,
"eval_samples_per_second": 15.769,
"eval_steps_per_second": 0.986,
"step": 56000
},
{
"epoch": 3.41,
"learning_rate": 3.2939791050184196e-05,
"loss": 0.4132,
"step": 56500
},
{
"epoch": 3.44,
"learning_rate": 3.278881574974335e-05,
"loss": 0.4118,
"step": 57000
},
{
"epoch": 3.47,
"learning_rate": 3.26378404493025e-05,
"loss": 0.3997,
"step": 57500
},
{
"epoch": 3.5,
"learning_rate": 3.248686514886165e-05,
"loss": 0.4088,
"step": 58000
},
{
"epoch": 3.53,
"learning_rate": 3.23358898484208e-05,
"loss": 0.3937,
"step": 58500
},
{
"epoch": 3.56,
"learning_rate": 3.218491454797995e-05,
"loss": 0.4001,
"step": 59000
},
{
"epoch": 3.59,
"learning_rate": 3.20339392475391e-05,
"loss": 0.3933,
"step": 59500
},
{
"epoch": 3.62,
"learning_rate": 3.1882963947098253e-05,
"loss": 0.3973,
"step": 60000
},
{
"epoch": 3.65,
"learning_rate": 3.1731988646657404e-05,
"loss": 0.4039,
"step": 60500
},
{
"epoch": 3.68,
"learning_rate": 3.158101334621656e-05,
"loss": 0.3913,
"step": 61000
},
{
"epoch": 3.71,
"learning_rate": 3.143003804577571e-05,
"loss": 0.3998,
"step": 61500
},
{
"epoch": 3.74,
"learning_rate": 3.1279062745334864e-05,
"loss": 0.392,
"step": 62000
},
{
"epoch": 3.77,
"learning_rate": 3.1128087444894015e-05,
"loss": 0.3931,
"step": 62500
},
{
"epoch": 3.8,
"learning_rate": 3.0977112144453166e-05,
"loss": 0.3859,
"step": 63000
},
{
"epoch": 3.83,
"learning_rate": 3.0826136844012324e-05,
"loss": 0.3826,
"step": 63500
},
{
"epoch": 3.86,
"learning_rate": 3.0675161543571475e-05,
"loss": 0.3852,
"step": 64000
},
{
"epoch": 3.86,
"eval_bleu": 8.5886,
"eval_gen_len": 17.7885,
"eval_loss": 1.0153683423995972,
"eval_runtime": 2110.3892,
"eval_samples_per_second": 15.693,
"eval_steps_per_second": 0.981,
"step": 64000
},
{
"epoch": 3.86,
"step": 64000,
"total_flos": 2.2192449043268895e+18,
"train_loss": 0.7053718340396881,
"train_runtime": 90544.7613,
"train_samples_per_second": 29.262,
"train_steps_per_second": 1.829
}
],
"logging_steps": 500,
"max_steps": 165590,
"num_train_epochs": 10,
"save_steps": 8000,
"total_flos": 2.2192449043268895e+18,
"trial_name": null,
"trial_params": null
}