File size: 4,945 Bytes
1b764dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 |
{
"best_metric": 0.6800228357315063,
"best_model_checkpoint": "/content/drive/MyDrive/Colab Notebooks/models/t5-small-codesearchnet-multilang-python-java/checkpoint-4125",
"epoch": 11.0,
"global_step": 4125,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_avg_length": 16.3976,
"eval_bleu": 0.0013,
"eval_loss": 0.9005016684532166,
"eval_rouge1": 0.1397,
"eval_rouge2": 0.0334,
"eval_runtime": 159.8289,
"eval_samples_per_second": 31.283,
"eval_steps_per_second": 3.91,
"step": 375
},
{
"epoch": 1.33,
"learning_rate": 0.011611906811594963,
"loss": 2.3568,
"step": 500
},
{
"epoch": 2.0,
"eval_avg_length": 15.8896,
"eval_bleu": 0.0023,
"eval_loss": 0.8036138415336609,
"eval_rouge1": 0.1737,
"eval_rouge2": 0.0526,
"eval_runtime": 164.0069,
"eval_samples_per_second": 30.487,
"eval_steps_per_second": 3.811,
"step": 750
},
{
"epoch": 2.67,
"learning_rate": 0.023264221847057343,
"loss": 0.7576,
"step": 1000
},
{
"epoch": 3.0,
"eval_avg_length": 15.3102,
"eval_bleu": 0.0021,
"eval_loss": 0.7583827972412109,
"eval_rouge1": 0.1856,
"eval_rouge2": 0.0558,
"eval_runtime": 167.7038,
"eval_samples_per_second": 29.814,
"eval_steps_per_second": 3.727,
"step": 1125
},
{
"epoch": 4.0,
"learning_rate": 0.03498469293117523,
"loss": 0.6778,
"step": 1500
},
{
"epoch": 4.0,
"eval_avg_length": 15.3544,
"eval_bleu": 0.0024,
"eval_loss": 0.7298200130462646,
"eval_rouge1": 0.1922,
"eval_rouge2": 0.0597,
"eval_runtime": 168.897,
"eval_samples_per_second": 29.604,
"eval_steps_per_second": 3.7,
"step": 1500
},
{
"epoch": 5.0,
"eval_avg_length": 15.7588,
"eval_bleu": 0.0037,
"eval_loss": 0.7113876938819885,
"eval_rouge1": 0.2114,
"eval_rouge2": 0.0704,
"eval_runtime": 172.1098,
"eval_samples_per_second": 29.051,
"eval_steps_per_second": 3.631,
"step": 1875
},
{
"epoch": 5.33,
"learning_rate": 0.046804994344711304,
"loss": 0.6206,
"step": 2000
},
{
"epoch": 6.0,
"eval_avg_length": 15.8088,
"eval_bleu": 0.0039,
"eval_loss": 0.6949090957641602,
"eval_rouge1": 0.2093,
"eval_rouge2": 0.0729,
"eval_runtime": 165.5542,
"eval_samples_per_second": 30.202,
"eval_steps_per_second": 3.775,
"step": 2250
},
{
"epoch": 6.67,
"learning_rate": 0.05876593291759491,
"loss": 0.5856,
"step": 2500
},
{
"epoch": 7.0,
"eval_avg_length": 16.5838,
"eval_bleu": 0.0042,
"eval_loss": 0.6927398443222046,
"eval_rouge1": 0.2143,
"eval_rouge2": 0.0711,
"eval_runtime": 164.6989,
"eval_samples_per_second": 30.358,
"eval_steps_per_second": 3.795,
"step": 2625
},
{
"epoch": 8.0,
"learning_rate": 0.07091041654348373,
"loss": 0.5447,
"step": 3000
},
{
"epoch": 8.0,
"eval_avg_length": 17.2174,
"eval_bleu": 0.005,
"eval_loss": 0.6866663694381714,
"eval_rouge1": 0.2151,
"eval_rouge2": 0.0717,
"eval_runtime": 163.5142,
"eval_samples_per_second": 30.578,
"eval_steps_per_second": 3.822,
"step": 3000
},
{
"epoch": 9.0,
"eval_avg_length": 16.1068,
"eval_bleu": 0.0043,
"eval_loss": 0.689474880695343,
"eval_rouge1": 0.2179,
"eval_rouge2": 0.0736,
"eval_runtime": 161.9004,
"eval_samples_per_second": 30.883,
"eval_steps_per_second": 3.86,
"step": 3375
},
{
"epoch": 9.33,
"learning_rate": 0.08324738591909409,
"loss": 0.5117,
"step": 3500
},
{
"epoch": 10.0,
"eval_avg_length": 15.5094,
"eval_bleu": 0.0038,
"eval_loss": 0.6876339912414551,
"eval_rouge1": 0.2229,
"eval_rouge2": 0.0777,
"eval_runtime": 168.6323,
"eval_samples_per_second": 29.65,
"eval_steps_per_second": 3.706,
"step": 3750
},
{
"epoch": 10.67,
"learning_rate": 0.09585673362016678,
"loss": 0.4892,
"step": 4000
},
{
"epoch": 11.0,
"eval_avg_length": 16.6902,
"eval_bleu": 0.0047,
"eval_loss": 0.6800228357315063,
"eval_rouge1": 0.2201,
"eval_rouge2": 0.0783,
"eval_runtime": 164.6191,
"eval_samples_per_second": 30.373,
"eval_steps_per_second": 3.797,
"step": 4125
}
],
"max_steps": 5625,
"num_train_epochs": 15,
"total_flos": 2.233139724288e+16,
"trial_name": null,
"trial_params": null
}
|