{ "best_metric": 4.724676132202148, "best_model_checkpoint": "/temp/mbart50-qg-aap/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.3892, "step": 100 }, { "epoch": 0.25, "learning_rate": 0.004752475247524752, "loss": 6.0843, "step": 200 }, { "epoch": 0.37, "learning_rate": 0.004628712871287129, "loss": 6.284, "step": 300 }, { "epoch": 0.5, "learning_rate": 0.0045049504950495055, "loss": 6.4282, "step": 400 }, { "epoch": 0.62, "learning_rate": 0.004381188118811882, "loss": 6.4175, "step": 500 }, { "epoch": 0.74, "learning_rate": 0.004257425742574258, "loss": 6.3385, "step": 600 }, { "epoch": 0.87, "learning_rate": 0.004133663366336634, "loss": 6.3127, "step": 700 }, { "epoch": 0.99, "learning_rate": 0.00400990099009901, "loss": 6.2697, "step": 800 }, { "epoch": 1.0, "eval_loss": 6.985348224639893, "eval_runtime": 204.6512, "eval_samples_per_second": 32.094, "eval_steps_per_second": 8.023, "step": 808 }, { "epoch": 1.11, "learning_rate": 0.0038861386138613866, "loss": 4.9723, "step": 900 }, { "epoch": 1.24, "learning_rate": 0.0037623762376237627, "loss": 4.9351, "step": 1000 }, { "epoch": 1.36, "learning_rate": 0.0036386138613861384, "loss": 4.9244, "step": 1100 }, { "epoch": 1.49, "learning_rate": 0.0035148514851485145, "loss": 4.9531, "step": 1200 }, { "epoch": 1.61, "learning_rate": 0.0033910891089108915, "loss": 5.0817, "step": 1300 }, { "epoch": 1.73, "learning_rate": 0.0032673267326732676, "loss": 5.0333, "step": 1400 }, { "epoch": 1.86, "learning_rate": 0.0031435643564356438, "loss": 4.9353, "step": 1500 }, { "epoch": 1.98, "learning_rate": 0.00301980198019802, "loss": 4.8922, "step": 1600 }, { "epoch": 2.0, "eval_loss": 6.513347148895264, "eval_runtime": 204.7144, "eval_samples_per_second": 32.084, "eval_steps_per_second": 8.021, "step": 1616 }, { "epoch": 2.1, "learning_rate": 0.002896039603960396, "loss": 4.0298, "step": 1700 }, { "epoch": 2.23, "learning_rate": 0.0027722772277227726, "loss": 3.8668, "step": 1800 }, { "epoch": 2.35, "learning_rate": 0.0026485148514851487, "loss": 3.8946, "step": 1900 }, { "epoch": 2.48, "learning_rate": 0.002524752475247525, "loss": 3.8475, "step": 2000 }, { "epoch": 2.6, "learning_rate": 0.002400990099009901, "loss": 3.8372, "step": 2100 }, { "epoch": 2.72, "learning_rate": 0.0022772277227722775, "loss": 3.7446, "step": 2200 }, { "epoch": 2.85, "learning_rate": 0.0021534653465346536, "loss": 3.7579, "step": 2300 }, { "epoch": 2.97, "learning_rate": 0.0020297029702970298, "loss": 3.7029, "step": 2400 }, { "epoch": 3.0, "eval_loss": 5.8359551429748535, "eval_runtime": 204.8301, "eval_samples_per_second": 32.066, "eval_steps_per_second": 8.016, "step": 2424 }, { "epoch": 3.09, "learning_rate": 0.0019059405940594061, "loss": 3.1314, "step": 2500 }, { "epoch": 3.22, "learning_rate": 0.001782178217821782, "loss": 2.9593, "step": 2600 }, { "epoch": 3.34, "learning_rate": 0.0016584158415841586, "loss": 2.9871, "step": 2700 }, { "epoch": 3.47, "learning_rate": 0.0015346534653465347, "loss": 2.959, "step": 2800 }, { "epoch": 3.59, "learning_rate": 0.0014108910891089108, "loss": 2.9037, "step": 2900 }, { "epoch": 3.71, "learning_rate": 0.0012871287128712872, "loss": 2.8455, "step": 3000 }, { "epoch": 3.84, "learning_rate": 0.0011633663366336635, "loss": 2.8385, "step": 3100 }, { "epoch": 3.96, "learning_rate": 0.0010396039603960397, "loss": 2.7608, "step": 3200 }, { "epoch": 4.0, "eval_loss": 5.15936279296875, "eval_runtime": 204.6859, "eval_samples_per_second": 32.088, "eval_steps_per_second": 8.022, "step": 3232 }, { "epoch": 4.08, "learning_rate": 0.0009158415841584158, "loss": 2.4672, "step": 3300 }, { "epoch": 4.21, "learning_rate": 0.0007920792079207921, "loss": 2.3107, "step": 3400 }, { "epoch": 4.33, "learning_rate": 0.0006683168316831684, "loss": 2.3218, "step": 3500 }, { "epoch": 4.46, "learning_rate": 0.0005445544554455446, "loss": 2.2424, "step": 3600 }, { "epoch": 4.58, "learning_rate": 0.00042079207920792084, "loss": 2.2134, "step": 3700 }, { "epoch": 4.7, "learning_rate": 0.000297029702970297, "loss": 2.1847, "step": 3800 }, { "epoch": 4.83, "learning_rate": 0.00017326732673267329, "loss": 2.1587, "step": 3900 }, { "epoch": 4.95, "learning_rate": 4.950495049504951e-05, "loss": 2.1148, "step": 4000 }, { "epoch": 5.0, "eval_loss": 4.724676132202148, "eval_runtime": 204.6885, "eval_samples_per_second": 32.088, "eval_steps_per_second": 8.022, "step": 4040 }, { "epoch": 5.0, "step": 4040, "total_flos": 3.6330926562607104e+17, "train_loss": 4.039364291653775, "train_runtime": 9395.5901, "train_samples_per_second": 27.515, "train_steps_per_second": 0.43 } ], "logging_steps": 100, "max_steps": 4040, "num_train_epochs": 5, "save_steps": 500, "total_flos": 3.6330926562607104e+17, "trial_name": null, "trial_params": null }