| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.999156296140055, | |
| "eval_steps": 500, | |
| "global_step": 9480, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 1.8955696202531647e-05, | |
| "loss": 2.9241, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 1.7900843881856543e-05, | |
| "loss": 2.7783, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 1.6845991561181435e-05, | |
| "loss": 2.7342, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "learning_rate": 1.579113924050633e-05, | |
| "loss": 2.6968, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_f1": 0.8701, | |
| "eval_gen_len": 19.4437588989084, | |
| "eval_loss": 2.538452386856079, | |
| "eval_precision": 0.8766, | |
| "eval_recall": 0.8641, | |
| "eval_rouge1": 0.2634, | |
| "eval_rouge2": 0.0907, | |
| "eval_rougeL": 0.218, | |
| "eval_rougeLsum": 0.2182, | |
| "eval_runtime": 339.9314, | |
| "eval_samples_per_second": 6.198, | |
| "eval_steps_per_second": 1.55, | |
| "step": 2370 | |
| }, | |
| { | |
| "epoch": 1.05, | |
| "learning_rate": 1.4736286919831224e-05, | |
| "loss": 2.6435, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 1.27, | |
| "learning_rate": 1.368354430379747e-05, | |
| "loss": 2.4844, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.48, | |
| "learning_rate": 1.2628691983122363e-05, | |
| "loss": 2.4626, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 1.69, | |
| "learning_rate": 1.1573839662447259e-05, | |
| "loss": 2.4676, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "learning_rate": 1.0523206751054854e-05, | |
| "loss": 2.4746, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_f1": 0.8712, | |
| "eval_gen_len": 19.25723777883246, | |
| "eval_loss": 2.507711410522461, | |
| "eval_precision": 0.8774, | |
| "eval_recall": 0.8655, | |
| "eval_rouge1": 0.273, | |
| "eval_rouge2": 0.0941, | |
| "eval_rougeL": 0.2238, | |
| "eval_rougeLsum": 0.2239, | |
| "eval_runtime": 335.7013, | |
| "eval_samples_per_second": 6.276, | |
| "eval_steps_per_second": 1.57, | |
| "step": 4741 | |
| }, | |
| { | |
| "epoch": 2.11, | |
| "learning_rate": 9.468354430379748e-06, | |
| "loss": 2.3813, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 2.32, | |
| "learning_rate": 8.415611814345991e-06, | |
| "loss": 2.3173, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 2.53, | |
| "learning_rate": 7.360759493670887e-06, | |
| "loss": 2.3064, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 2.74, | |
| "learning_rate": 6.305907172995781e-06, | |
| "loss": 2.3008, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 2.95, | |
| "learning_rate": 5.251054852320675e-06, | |
| "loss": 2.3066, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_f1": 0.8696, | |
| "eval_gen_len": 19.307071665875654, | |
| "eval_loss": 2.501197099685669, | |
| "eval_precision": 0.8756, | |
| "eval_recall": 0.864, | |
| "eval_rouge1": 0.2671, | |
| "eval_rouge2": 0.0936, | |
| "eval_rougeL": 0.221, | |
| "eval_rougeLsum": 0.2211, | |
| "eval_runtime": 336.0052, | |
| "eval_samples_per_second": 6.271, | |
| "eval_steps_per_second": 1.568, | |
| "step": 7111 | |
| }, | |
| { | |
| "epoch": 3.16, | |
| "learning_rate": 4.19620253164557e-06, | |
| "loss": 2.2216, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 3.37, | |
| "learning_rate": 3.141350210970465e-06, | |
| "loss": 2.2071, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 3.59, | |
| "learning_rate": 2.0864978902953587e-06, | |
| "loss": 2.1852, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 3.8, | |
| "learning_rate": 1.0316455696202532e-06, | |
| "loss": 2.2041, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_f1": 0.8705, | |
| "eval_gen_len": 19.35548172757475, | |
| "eval_loss": 2.503497838973999, | |
| "eval_precision": 0.8768, | |
| "eval_recall": 0.8648, | |
| "eval_rouge1": 0.2709, | |
| "eval_rouge2": 0.0948, | |
| "eval_rougeL": 0.2244, | |
| "eval_rougeLsum": 0.2244, | |
| "eval_runtime": 336.6782, | |
| "eval_samples_per_second": 6.258, | |
| "eval_steps_per_second": 1.565, | |
| "step": 9480 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "step": 9480, | |
| "total_flos": 2.372598264346706e+17, | |
| "train_loss": 2.436967770862177, | |
| "train_runtime": 10313.084, | |
| "train_samples_per_second": 14.709, | |
| "train_steps_per_second": 0.919 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 9480, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 4, | |
| "save_steps": 500, | |
| "total_flos": 2.372598264346706e+17, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |