| { | |
| "best_metric": 4.769406318664551, | |
| "best_model_checkpoint": "/temp/mbart50-qg-af/checkpoint-4040", | |
| "epoch": 5.0, | |
| "eval_steps": 500, | |
| "global_step": 4040, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 0.004876237623762376, | |
| "loss": 6.3971, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 0.004752475247524752, | |
| "loss": 6.1148, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 0.004628712871287129, | |
| "loss": 6.3394, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "learning_rate": 0.0045049504950495055, | |
| "loss": 6.4336, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 0.004381188118811882, | |
| "loss": 6.3685, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "learning_rate": 0.004257425742574258, | |
| "loss": 6.3731, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "learning_rate": 0.004133663366336634, | |
| "loss": 6.3788, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "learning_rate": 0.00400990099009901, | |
| "loss": 6.3227, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_loss": 7.025113105773926, | |
| "eval_runtime": 205.4905, | |
| "eval_samples_per_second": 31.928, | |
| "eval_steps_per_second": 7.986, | |
| "step": 808 | |
| }, | |
| { | |
| "epoch": 1.11, | |
| "learning_rate": 0.0038861386138613866, | |
| "loss": 5.0392, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 1.24, | |
| "learning_rate": 0.0037623762376237627, | |
| "loss": 4.967, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.36, | |
| "learning_rate": 0.0036386138613861384, | |
| "loss": 4.9332, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.49, | |
| "learning_rate": 0.0035148514851485145, | |
| "loss": 4.9914, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.61, | |
| "learning_rate": 0.0033910891089108915, | |
| "loss": 5.0841, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.73, | |
| "learning_rate": 0.0032673267326732676, | |
| "loss": 5.0434, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.86, | |
| "learning_rate": 0.0031435643564356438, | |
| "loss": 4.9521, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.98, | |
| "learning_rate": 0.00301980198019802, | |
| "loss": 4.8835, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_loss": 6.693631172180176, | |
| "eval_runtime": 205.2634, | |
| "eval_samples_per_second": 31.964, | |
| "eval_steps_per_second": 7.995, | |
| "step": 1616 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "learning_rate": 0.002896039603960396, | |
| "loss": 4.0058, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 2.23, | |
| "learning_rate": 0.0027722772277227726, | |
| "loss": 3.8947, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 2.35, | |
| "learning_rate": 0.0026485148514851487, | |
| "loss": 3.9128, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 2.48, | |
| "learning_rate": 0.002524752475247525, | |
| "loss": 3.8698, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.6, | |
| "learning_rate": 0.002400990099009901, | |
| "loss": 3.867, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.72, | |
| "learning_rate": 0.0022772277227722775, | |
| "loss": 3.7577, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.85, | |
| "learning_rate": 0.0021534653465346536, | |
| "loss": 3.7689, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "learning_rate": 0.0020297029702970298, | |
| "loss": 3.7212, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_loss": 5.869358062744141, | |
| "eval_runtime": 205.0568, | |
| "eval_samples_per_second": 31.996, | |
| "eval_steps_per_second": 8.003, | |
| "step": 2424 | |
| }, | |
| { | |
| "epoch": 3.09, | |
| "learning_rate": 0.0019059405940594061, | |
| "loss": 3.1567, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 3.22, | |
| "learning_rate": 0.001782178217821782, | |
| "loss": 2.9952, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 3.34, | |
| "learning_rate": 0.0016584158415841586, | |
| "loss": 2.9908, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 3.47, | |
| "learning_rate": 0.0015346534653465347, | |
| "loss": 2.9718, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 3.59, | |
| "learning_rate": 0.0014108910891089108, | |
| "loss": 2.9438, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 3.71, | |
| "learning_rate": 0.0012871287128712872, | |
| "loss": 2.8563, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.84, | |
| "learning_rate": 0.0011633663366336635, | |
| "loss": 2.8644, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 3.96, | |
| "learning_rate": 0.0010396039603960397, | |
| "loss": 2.7979, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_loss": 5.221120834350586, | |
| "eval_runtime": 204.9921, | |
| "eval_samples_per_second": 32.006, | |
| "eval_steps_per_second": 8.005, | |
| "step": 3232 | |
| }, | |
| { | |
| "epoch": 4.08, | |
| "learning_rate": 0.0009158415841584158, | |
| "loss": 2.4949, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 4.21, | |
| "learning_rate": 0.0007920792079207921, | |
| "loss": 2.3339, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 4.33, | |
| "learning_rate": 0.0006683168316831684, | |
| "loss": 2.3368, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 4.46, | |
| "learning_rate": 0.0005445544554455446, | |
| "loss": 2.2803, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 4.58, | |
| "learning_rate": 0.00042079207920792084, | |
| "loss": 2.2278, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 4.7, | |
| "learning_rate": 0.000297029702970297, | |
| "loss": 2.2057, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 4.83, | |
| "learning_rate": 0.00017326732673267329, | |
| "loss": 2.1881, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 4.95, | |
| "learning_rate": 4.950495049504951e-05, | |
| "loss": 2.14, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_loss": 4.769406318664551, | |
| "eval_runtime": 204.8793, | |
| "eval_samples_per_second": 32.024, | |
| "eval_steps_per_second": 8.01, | |
| "step": 4040 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "step": 4040, | |
| "total_flos": 3.6330926562607104e+17, | |
| "train_loss": 4.061046307629878, | |
| "train_runtime": 9303.8582, | |
| "train_samples_per_second": 27.786, | |
| "train_steps_per_second": 0.434 | |
| } | |
| ], | |
| "logging_steps": 100, | |
| "max_steps": 4040, | |
| "num_train_epochs": 5, | |
| "save_steps": 500, | |
| "total_flos": 3.6330926562607104e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |