File size: 5,598 Bytes
8309710 | 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 198 199 200 201 202 203 204 205 206 207 208 209 210 211 | {
"best_metric": 2.276757001876831,
"best_model_checkpoint": "tiny_bert_rand_100_v1_stsb/checkpoint-115",
"epoch": 10.0,
"eval_steps": 500,
"global_step": 230,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 4.961604595184326,
"learning_rate": 4.9e-05,
"loss": 3.4483,
"step": 23
},
{
"epoch": 1.0,
"eval_combined_score": 0.15705636987694221,
"eval_loss": 2.314561128616333,
"eval_pearson": 0.16774334792257545,
"eval_runtime": 0.5294,
"eval_samples_per_second": 2833.424,
"eval_spearmanr": 0.14636939183130895,
"eval_steps_per_second": 11.334,
"step": 23
},
{
"epoch": 2.0,
"grad_norm": 6.913020610809326,
"learning_rate": 4.8e-05,
"loss": 2.0255,
"step": 46
},
{
"epoch": 2.0,
"eval_combined_score": 0.1126483992024648,
"eval_loss": 2.5450494289398193,
"eval_pearson": 0.11683712415863884,
"eval_runtime": 0.5808,
"eval_samples_per_second": 2582.785,
"eval_spearmanr": 0.10845967424629079,
"eval_steps_per_second": 10.331,
"step": 46
},
{
"epoch": 3.0,
"grad_norm": 7.117321491241455,
"learning_rate": 4.7e-05,
"loss": 1.8523,
"step": 69
},
{
"epoch": 3.0,
"eval_combined_score": 0.21422209468092734,
"eval_loss": 2.3148019313812256,
"eval_pearson": 0.2201949812893677,
"eval_runtime": 0.6512,
"eval_samples_per_second": 2303.455,
"eval_spearmanr": 0.20824920807248695,
"eval_steps_per_second": 9.214,
"step": 69
},
{
"epoch": 4.0,
"grad_norm": 6.040245532989502,
"learning_rate": 4.600000000000001e-05,
"loss": 1.6156,
"step": 92
},
{
"epoch": 4.0,
"eval_combined_score": 0.2691090690581144,
"eval_loss": 2.3426806926727295,
"eval_pearson": 0.27034978222621336,
"eval_runtime": 0.548,
"eval_samples_per_second": 2737.269,
"eval_spearmanr": 0.26786835589001545,
"eval_steps_per_second": 10.949,
"step": 92
},
{
"epoch": 5.0,
"grad_norm": 9.559501647949219,
"learning_rate": 4.5e-05,
"loss": 1.3454,
"step": 115
},
{
"epoch": 5.0,
"eval_combined_score": 0.2780937290708646,
"eval_loss": 2.276757001876831,
"eval_pearson": 0.2798131537425998,
"eval_runtime": 0.5306,
"eval_samples_per_second": 2827.107,
"eval_spearmanr": 0.2763743043991294,
"eval_steps_per_second": 11.308,
"step": 115
},
{
"epoch": 6.0,
"grad_norm": 7.285312652587891,
"learning_rate": 4.4000000000000006e-05,
"loss": 1.1616,
"step": 138
},
{
"epoch": 6.0,
"eval_combined_score": 0.273431489839639,
"eval_loss": 2.6383583545684814,
"eval_pearson": 0.26860800836532117,
"eval_runtime": 0.5919,
"eval_samples_per_second": 2534.01,
"eval_spearmanr": 0.2782549713139568,
"eval_steps_per_second": 10.136,
"step": 138
},
{
"epoch": 7.0,
"grad_norm": 10.173330307006836,
"learning_rate": 4.3e-05,
"loss": 0.9734,
"step": 161
},
{
"epoch": 7.0,
"eval_combined_score": 0.2831437139838975,
"eval_loss": 2.477155923843384,
"eval_pearson": 0.28229446668730734,
"eval_runtime": 0.5315,
"eval_samples_per_second": 2822.41,
"eval_spearmanr": 0.2839929612804877,
"eval_steps_per_second": 11.29,
"step": 161
},
{
"epoch": 8.0,
"grad_norm": 9.35107707977295,
"learning_rate": 4.2e-05,
"loss": 0.8406,
"step": 184
},
{
"epoch": 8.0,
"eval_combined_score": 0.24872085472419414,
"eval_loss": 2.8825998306274414,
"eval_pearson": 0.2434565004022157,
"eval_runtime": 0.5698,
"eval_samples_per_second": 2632.351,
"eval_spearmanr": 0.25398520904617256,
"eval_steps_per_second": 10.529,
"step": 184
},
{
"epoch": 9.0,
"grad_norm": 17.66458511352539,
"learning_rate": 4.1e-05,
"loss": 0.7077,
"step": 207
},
{
"epoch": 9.0,
"eval_combined_score": 0.24927091603402246,
"eval_loss": 2.909149169921875,
"eval_pearson": 0.2461040064609805,
"eval_runtime": 0.5293,
"eval_samples_per_second": 2834.011,
"eval_spearmanr": 0.25243782560706446,
"eval_steps_per_second": 11.336,
"step": 207
},
{
"epoch": 10.0,
"grad_norm": 11.106776237487793,
"learning_rate": 4e-05,
"loss": 0.6149,
"step": 230
},
{
"epoch": 10.0,
"eval_combined_score": 0.2685087682776196,
"eval_loss": 2.823472023010254,
"eval_pearson": 0.26516905911909416,
"eval_runtime": 0.5621,
"eval_samples_per_second": 2668.438,
"eval_spearmanr": 0.2718484774361451,
"eval_steps_per_second": 10.674,
"step": 230
},
{
"epoch": 10.0,
"step": 230,
"total_flos": 1507548205378560.0,
"train_loss": 1.4585458755493164,
"train_runtime": 51.1618,
"train_samples_per_second": 5618.449,
"train_steps_per_second": 22.478
}
],
"logging_steps": 1,
"max_steps": 1150,
"num_input_tokens_seen": 0,
"num_train_epochs": 50,
"save_steps": 500,
"total_flos": 1507548205378560.0,
"train_batch_size": 256,
"trial_name": null,
"trial_params": null
}
|