File size: 4,933 Bytes
2cb4256 |
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 |
{
"best_metric": 0.21319285035133362,
"best_model_checkpoint": "./mms-1b-bem-female-sv/checkpoint-2400",
"epoch": 5.0,
"eval_steps": 200,
"global_step": 2505,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.3992015968063872,
"eval_loss": 0.3566432595252991,
"eval_runtime": 69.3187,
"eval_samples_per_second": 6.852,
"eval_steps_per_second": 0.866,
"eval_wer": 0.5019098548510313,
"step": 200
},
{
"epoch": 0.7984031936127745,
"eval_loss": 0.26201513409614563,
"eval_runtime": 68.6891,
"eval_samples_per_second": 6.915,
"eval_steps_per_second": 0.874,
"eval_wer": 0.4028520499108734,
"step": 400
},
{
"epoch": 0.998003992015968,
"grad_norm": 1.1227352619171143,
"learning_rate": 0.000994,
"loss": 1.7214,
"step": 500
},
{
"epoch": 1.1976047904191618,
"eval_loss": 0.2546118199825287,
"eval_runtime": 69.4229,
"eval_samples_per_second": 6.842,
"eval_steps_per_second": 0.864,
"eval_wer": 0.40998217468805703,
"step": 600
},
{
"epoch": 1.596806387225549,
"eval_loss": 0.2358996421098709,
"eval_runtime": 68.6925,
"eval_samples_per_second": 6.915,
"eval_steps_per_second": 0.873,
"eval_wer": 0.3964858670741024,
"step": 800
},
{
"epoch": 1.996007984031936,
"grad_norm": 2.2609593868255615,
"learning_rate": 0.0007531172069825436,
"loss": 0.2801,
"step": 1000
},
{
"epoch": 1.996007984031936,
"eval_loss": 0.23218725621700287,
"eval_runtime": 68.6733,
"eval_samples_per_second": 6.917,
"eval_steps_per_second": 0.874,
"eval_wer": 0.38095238095238093,
"step": 1000
},
{
"epoch": 2.3952095808383236,
"eval_loss": 0.23050391674041748,
"eval_runtime": 68.6828,
"eval_samples_per_second": 6.916,
"eval_steps_per_second": 0.874,
"eval_wer": 0.3745861981156099,
"step": 1200
},
{
"epoch": 2.7944111776447107,
"eval_loss": 0.22583143413066864,
"eval_runtime": 68.7708,
"eval_samples_per_second": 6.907,
"eval_steps_per_second": 0.872,
"eval_wer": 0.33358798064680417,
"step": 1400
},
{
"epoch": 2.9940119760479043,
"grad_norm": 0.9439337849617004,
"learning_rate": 0.0005037406483790524,
"loss": 0.2528,
"step": 1500
},
{
"epoch": 3.193612774451098,
"eval_loss": 0.22624839842319489,
"eval_runtime": 68.8386,
"eval_samples_per_second": 6.9,
"eval_steps_per_second": 0.872,
"eval_wer": 0.43086325439266615,
"step": 1600
},
{
"epoch": 3.592814371257485,
"eval_loss": 0.2164432853460312,
"eval_runtime": 68.775,
"eval_samples_per_second": 6.907,
"eval_steps_per_second": 0.872,
"eval_wer": 0.35141329258976317,
"step": 1800
},
{
"epoch": 3.992015968063872,
"grad_norm": 1.3000344038009644,
"learning_rate": 0.0002543640897755611,
"loss": 0.2351,
"step": 2000
},
{
"epoch": 3.992015968063872,
"eval_loss": 0.22146381437778473,
"eval_runtime": 68.601,
"eval_samples_per_second": 6.924,
"eval_steps_per_second": 0.875,
"eval_wer": 0.38935574229691877,
"step": 2000
},
{
"epoch": 4.391217564870259,
"eval_loss": 0.21647760272026062,
"eval_runtime": 68.6439,
"eval_samples_per_second": 6.92,
"eval_steps_per_second": 0.874,
"eval_wer": 0.3623631270690094,
"step": 2200
},
{
"epoch": 4.790419161676647,
"eval_loss": 0.21319285035133362,
"eval_runtime": 68.6858,
"eval_samples_per_second": 6.916,
"eval_steps_per_second": 0.874,
"eval_wer": 0.3557422969187675,
"step": 2400
},
{
"epoch": 4.99001996007984,
"grad_norm": 14.332290649414062,
"learning_rate": 4.987531172069825e-06,
"loss": 0.2222,
"step": 2500
},
{
"epoch": 5.0,
"step": 2505,
"total_flos": 1.1971485322227634e+19,
"train_loss": 0.5417278716188229,
"train_runtime": 6598.5921,
"train_samples_per_second": 3.036,
"train_steps_per_second": 0.38
}
],
"logging_steps": 500,
"max_steps": 2505,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 200,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 1.1971485322227634e+19,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}
|