{ "best_metric": 0.7699160575866699, "best_model_checkpoint": "/home2/s5431786/jb-internship/results/Salesforce-codet5-small-CodeXGLUE-CONCODE-selected/checkpoint-8000", "epoch": 3.2, "global_step": 10000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.64, "learning_rate": 6.3808e-05, "loss": 1.45, "step": 2000 }, { "epoch": 0.64, "eval_bleu": 0.24715996940066084, "eval_exact_match": 0.1615, "eval_loss": 0.8543930649757385, "eval_rouge1": 0.5607398648042461, "eval_rouge2": 0.376714337460786, "eval_rougeL": 0.5368883844476564, "eval_runtime": 233.0119, "eval_samples_per_second": 8.583, "eval_steps_per_second": 0.27, "step": 2000 }, { "epoch": 1.28, "learning_rate": 9.853642105263158e-05, "loss": 0.7186, "step": 4000 }, { "epoch": 1.28, "eval_bleu": 0.2686100325072973, "eval_exact_match": 0.1755, "eval_loss": 0.8113107681274414, "eval_rouge1": 0.5759137545878418, "eval_rouge2": 0.3934978008929957, "eval_rougeL": 0.5514257399531202, "eval_runtime": 222.3222, "eval_samples_per_second": 8.996, "eval_steps_per_second": 0.283, "step": 4000 }, { "epoch": 1.92, "learning_rate": 9.516800000000001e-05, "loss": 0.6385, "step": 6000 }, { "epoch": 1.92, "eval_bleu": 0.2859251068334253, "eval_exact_match": 0.1795, "eval_loss": 0.7783799767494202, "eval_rouge1": 0.5758903457879601, "eval_rouge2": 0.4004382979016302, "eval_rougeL": 0.5524759373364955, "eval_runtime": 259.7789, "eval_samples_per_second": 7.699, "eval_steps_per_second": 0.243, "step": 6000 }, { "epoch": 2.56, "learning_rate": 9.179957894736842e-05, "loss": 0.562, "step": 8000 }, { "epoch": 2.56, "eval_bleu": 0.2967822969771582, "eval_exact_match": 0.179, "eval_loss": 0.7699160575866699, "eval_rouge1": 0.5769004117490721, "eval_rouge2": 0.400017695314681, "eval_rougeL": 0.5526377939756941, "eval_runtime": 274.6962, "eval_samples_per_second": 7.281, "eval_steps_per_second": 0.229, "step": 8000 }, { "epoch": 3.2, "learning_rate": 8.843115789473684e-05, "loss": 0.5314, "step": 10000 }, { "epoch": 3.2, "eval_bleu": 0.29975143100679225, "eval_exact_match": 0.1815, "eval_loss": 0.7741894125938416, "eval_rouge1": 0.595198816050763, "eval_rouge2": 0.4127741738801316, "eval_rougeL": 0.5696024067817068, "eval_runtime": 220.4946, "eval_samples_per_second": 9.071, "eval_steps_per_second": 0.286, "step": 10000 } ], "max_steps": 62500, "num_train_epochs": 20, "total_flos": 3.612457930378445e+16, "trial_name": null, "trial_params": null }