| { | |
| "best_metric": 1.528552770614624, | |
| "best_model_checkpoint": "/home2/s5431786/jb-internship-repr/results/Salesforce-codet5-small-java-small-selected-wo-tokens/checkpoint-2000", | |
| "epoch": 1.9420169232903315, | |
| "global_step": 14000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 9.94363646983237e-05, | |
| "loss": 2.1009, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "eval_exact_match": 0.31785774198959904, | |
| "eval_loss": 1.528552770614624, | |
| "eval_rouge1": 0.5232290073169683, | |
| "eval_rouge2": 0.24795642157242634, | |
| "eval_rougeL": 0.5233033679411545, | |
| "eval_runtime": 163.446, | |
| "eval_samples_per_second": 145.883, | |
| "eval_steps_per_second": 1.523, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 9.359558437007184e-05, | |
| "loss": 1.3098, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "eval_exact_match": 0.29667840966280823, | |
| "eval_loss": 1.5582195520401, | |
| "eval_rouge1": 0.499918373090859, | |
| "eval_rouge2": 0.22982882883486827, | |
| "eval_rougeL": 0.49974542584946463, | |
| "eval_runtime": 169.5108, | |
| "eval_samples_per_second": 140.664, | |
| "eval_steps_per_second": 1.469, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 8.775480404182e-05, | |
| "loss": 1.1968, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "eval_exact_match": 0.321003187384667, | |
| "eval_loss": 1.551592469215393, | |
| "eval_rouge1": 0.5260578932396776, | |
| "eval_rouge2": 0.25462429236963685, | |
| "eval_rougeL": 0.5257669098334219, | |
| "eval_runtime": 163.9432, | |
| "eval_samples_per_second": 145.441, | |
| "eval_steps_per_second": 1.519, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 1.11, | |
| "learning_rate": 8.191402371356814e-05, | |
| "loss": 1.1145, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 1.11, | |
| "eval_exact_match": 0.31735447072638817, | |
| "eval_loss": 1.572016954421997, | |
| "eval_rouge1": 0.5094747473558878, | |
| "eval_rouge2": 0.2465356707776194, | |
| "eval_rougeL": 0.5091592701411148, | |
| "eval_runtime": 163.415, | |
| "eval_samples_per_second": 145.911, | |
| "eval_steps_per_second": 1.524, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "learning_rate": 7.607908416564453e-05, | |
| "loss": 1.0477, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "eval_exact_match": 0.33387854386847843, | |
| "eval_loss": 1.5617761611938477, | |
| "eval_rouge1": 0.5280223151246797, | |
| "eval_rouge2": 0.2615585801921996, | |
| "eval_rougeL": 0.5274734242132477, | |
| "eval_runtime": 165.805, | |
| "eval_samples_per_second": 143.807, | |
| "eval_steps_per_second": 1.502, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "learning_rate": 7.023830383739268e-05, | |
| "loss": 1.0219, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "eval_exact_match": 0.32372924006039255, | |
| "eval_loss": 1.5944069623947144, | |
| "eval_rouge1": 0.5133412941637794, | |
| "eval_rouge2": 0.24304376784560536, | |
| "eval_rougeL": 0.5128045498112566, | |
| "eval_runtime": 164.4858, | |
| "eval_samples_per_second": 144.961, | |
| "eval_steps_per_second": 1.514, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 1.94, | |
| "learning_rate": 6.440044389930495e-05, | |
| "loss": 1.0027, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 1.94, | |
| "eval_exact_match": 0.3238550578761953, | |
| "eval_loss": 1.5750421285629272, | |
| "eval_rouge1": 0.5206336041126007, | |
| "eval_rouge2": 0.24470678869584325, | |
| "eval_rougeL": 0.5202257107678057, | |
| "eval_runtime": 161.6161, | |
| "eval_samples_per_second": 147.535, | |
| "eval_steps_per_second": 1.541, | |
| "step": 14000 | |
| } | |
| ], | |
| "max_steps": 36045, | |
| "num_train_epochs": 5, | |
| "total_flos": 1.8188720041623552e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |