{ "best_metric": 15.5769, "best_model_checkpoint": "saved/tobyoki-pairwise-wo_space/bart-base-japanese/BaseModel/checkpoint-2868", "epoch": 5.0, "global_step": 3585, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.7, "learning_rate": 4.302649930264993e-05, "loss": 2.1701, "step": 500 }, { "epoch": 1.0, "eval_gen_len": 47.7375, "eval_loss": 1.9507354497909546, "eval_rouge1": 12.6467, "eval_rouge2": 2.901, "eval_rougeL": 10.0035, "eval_rougeLsum": 11.1471, "eval_runtime": 40.6217, "eval_samples_per_second": 1.969, "eval_steps_per_second": 1.969, "step": 717 }, { "epoch": 1.39, "learning_rate": 3.6052998605299864e-05, "loss": 1.4042, "step": 1000 }, { "epoch": 2.0, "eval_gen_len": 21.1375, "eval_loss": 1.9519145488739014, "eval_rouge1": 11.9515, "eval_rouge2": 3.096, "eval_rougeL": 10.2259, "eval_rougeLsum": 10.8478, "eval_runtime": 26.1922, "eval_samples_per_second": 3.054, "eval_steps_per_second": 3.054, "step": 1434 }, { "epoch": 2.09, "learning_rate": 2.9079497907949792e-05, "loss": 1.1417, "step": 1500 }, { "epoch": 2.79, "learning_rate": 2.2105997210599723e-05, "loss": 0.8952, "step": 2000 }, { "epoch": 3.0, "eval_gen_len": 76.35, "eval_loss": 2.0322935581207275, "eval_rouge1": 15.5721, "eval_rouge2": 3.5875, "eval_rougeL": 10.6382, "eval_rougeLsum": 12.9346, "eval_runtime": 74.4084, "eval_samples_per_second": 1.075, "eval_steps_per_second": 1.075, "step": 2151 }, { "epoch": 3.49, "learning_rate": 1.5132496513249652e-05, "loss": 0.7489, "step": 2500 }, { "epoch": 4.0, "eval_gen_len": 63.7625, "eval_loss": 2.0723605155944824, "eval_rouge1": 15.5769, "eval_rouge2": 3.3042, "eval_rougeL": 11.0176, "eval_rougeLsum": 12.8107, "eval_runtime": 64.4381, "eval_samples_per_second": 1.242, "eval_steps_per_second": 1.242, "step": 2868 }, { "epoch": 4.18, "learning_rate": 8.158995815899583e-06, "loss": 0.6757, "step": 3000 }, { "epoch": 4.88, "learning_rate": 1.185495118549512e-06, "loss": 0.5941, "step": 3500 }, { "epoch": 5.0, "eval_gen_len": 66.075, "eval_loss": 2.0849809646606445, "eval_rouge1": 15.3454, "eval_rouge2": 2.9489, "eval_rougeL": 10.7691, "eval_rougeLsum": 12.7028, "eval_runtime": 57.1206, "eval_samples_per_second": 1.401, "eval_steps_per_second": 1.401, "step": 3585 }, { "epoch": 5.0, "step": 3585, "total_flos": 682678854881280.0, "train_loss": 1.0774688316355856, "train_runtime": 782.0878, "train_samples_per_second": 4.584, "train_steps_per_second": 4.584 } ], "max_steps": 3585, "num_train_epochs": 5, "total_flos": 682678854881280.0, "trial_name": null, "trial_params": null }