|
{ |
|
"best_metric": 17.98756798756799, |
|
"best_model_checkpoint": "./whisper-small-vivos/checkpoint-3000", |
|
"epoch": 4.11522633744856, |
|
"global_step": 3000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 4.6000000000000004e-07, |
|
"loss": 3.0428, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 9.600000000000001e-07, |
|
"loss": 2.6303, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 1.46e-06, |
|
"loss": 2.0752, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 1.9600000000000003e-06, |
|
"loss": 1.5733, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 2.46e-06, |
|
"loss": 1.1017, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 2.96e-06, |
|
"loss": 0.9, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 3.46e-06, |
|
"loss": 0.7899, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 3.96e-06, |
|
"loss": 0.6859, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.4600000000000005e-06, |
|
"loss": 0.6151, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.960000000000001e-06, |
|
"loss": 0.539, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 5.460000000000001e-06, |
|
"loss": 0.37, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 5.9600000000000005e-06, |
|
"loss": 0.3106, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 6.460000000000001e-06, |
|
"loss": 0.2854, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 6.96e-06, |
|
"loss": 0.2651, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 7.4600000000000006e-06, |
|
"loss": 0.2473, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 7.960000000000002e-06, |
|
"loss": 0.2256, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 8.46e-06, |
|
"loss": 0.2195, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 8.96e-06, |
|
"loss": 0.2266, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 9.460000000000001e-06, |
|
"loss": 0.216, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 9.960000000000001e-06, |
|
"loss": 0.2294, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1918, |
|
"step": 525 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 1e-05, |
|
"loss": 0.2152, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 1e-05, |
|
"loss": 0.187, |
|
"step": 575 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1756, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1862, |
|
"step": 625 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1744, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1764, |
|
"step": 675 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1662, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1788, |
|
"step": 725 |
|
}, |
|
{ |
|
"epoch": 1.03, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1472, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 1.06, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1381, |
|
"step": 775 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1277, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"learning_rate": 1e-05, |
|
"loss": 0.13, |
|
"step": 825 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1309, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1279, |
|
"step": 875 |
|
}, |
|
{ |
|
"epoch": 1.23, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1221, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 1.27, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1263, |
|
"step": 925 |
|
}, |
|
{ |
|
"epoch": 1.3, |
|
"learning_rate": 1e-05, |
|
"loss": 0.122, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 1.34, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1251, |
|
"step": 975 |
|
}, |
|
{ |
|
"epoch": 1.37, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1276, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.37, |
|
"eval_cer": 8.6875591223948, |
|
"eval_loss": 0.21370840072631836, |
|
"eval_runtime": 393.7923, |
|
"eval_samples_per_second": 1.93, |
|
"eval_steps_per_second": 0.122, |
|
"eval_wer": 20.616420616420616, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.41, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1184, |
|
"step": 1025 |
|
}, |
|
{ |
|
"epoch": 1.44, |
|
"learning_rate": 1e-05, |
|
"loss": 0.125, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 1.47, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1163, |
|
"step": 1075 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1263, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 1.54, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1129, |
|
"step": 1125 |
|
}, |
|
{ |
|
"epoch": 1.58, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1199, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 1.61, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1231, |
|
"step": 1175 |
|
}, |
|
{ |
|
"epoch": 1.65, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1195, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 1.68, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1145, |
|
"step": 1225 |
|
}, |
|
{ |
|
"epoch": 1.71, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1115, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 1.75, |
|
"learning_rate": 1e-05, |
|
"loss": 0.117, |
|
"step": 1275 |
|
}, |
|
{ |
|
"epoch": 1.78, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1089, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 1.82, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1143, |
|
"step": 1325 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1195, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1137, |
|
"step": 1375 |
|
}, |
|
{ |
|
"epoch": 1.92, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1142, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 1.95, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1022, |
|
"step": 1425 |
|
}, |
|
{ |
|
"epoch": 1.99, |
|
"learning_rate": 1e-05, |
|
"loss": 0.1, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0854, |
|
"step": 1475 |
|
}, |
|
{ |
|
"epoch": 2.06, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0707, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 2.09, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0702, |
|
"step": 1525 |
|
}, |
|
{ |
|
"epoch": 2.13, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0647, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 2.16, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0713, |
|
"step": 1575 |
|
}, |
|
{ |
|
"epoch": 2.19, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0627, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 2.23, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0678, |
|
"step": 1625 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0667, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 2.3, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0795, |
|
"step": 1675 |
|
}, |
|
{ |
|
"epoch": 2.33, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0608, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 2.37, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0642, |
|
"step": 1725 |
|
}, |
|
{ |
|
"epoch": 2.4, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0684, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 2.43, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0676, |
|
"step": 1775 |
|
}, |
|
{ |
|
"epoch": 2.47, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0743, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 2.5, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0713, |
|
"step": 1825 |
|
}, |
|
{ |
|
"epoch": 2.54, |
|
"learning_rate": 1e-05, |
|
"loss": 0.062, |
|
"step": 1850 |
|
}, |
|
{ |
|
"epoch": 2.57, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0678, |
|
"step": 1875 |
|
}, |
|
{ |
|
"epoch": 2.61, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0668, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 2.64, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0739, |
|
"step": 1925 |
|
}, |
|
{ |
|
"epoch": 2.67, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0672, |
|
"step": 1950 |
|
}, |
|
{ |
|
"epoch": 2.71, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0653, |
|
"step": 1975 |
|
}, |
|
{ |
|
"epoch": 2.74, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0696, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 2.74, |
|
"eval_cer": 7.509688444051142, |
|
"eval_loss": 0.217300683259964, |
|
"eval_runtime": 392.9774, |
|
"eval_samples_per_second": 1.934, |
|
"eval_steps_per_second": 0.122, |
|
"eval_wer": 19.08831908831909, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 2.78, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0816, |
|
"step": 2025 |
|
}, |
|
{ |
|
"epoch": 2.81, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0796, |
|
"step": 2050 |
|
}, |
|
{ |
|
"epoch": 2.85, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0721, |
|
"step": 2075 |
|
}, |
|
{ |
|
"epoch": 2.88, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0716, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 2.91, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0659, |
|
"step": 2125 |
|
}, |
|
{ |
|
"epoch": 2.95, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0644, |
|
"step": 2150 |
|
}, |
|
{ |
|
"epoch": 2.98, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0653, |
|
"step": 2175 |
|
}, |
|
{ |
|
"epoch": 3.02, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0558, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"learning_rate": 1e-05, |
|
"loss": 0.043, |
|
"step": 2225 |
|
}, |
|
{ |
|
"epoch": 3.09, |
|
"learning_rate": 1e-05, |
|
"loss": 0.047, |
|
"step": 2250 |
|
}, |
|
{ |
|
"epoch": 3.12, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0355, |
|
"step": 2275 |
|
}, |
|
{ |
|
"epoch": 3.16, |
|
"learning_rate": 1e-05, |
|
"loss": 0.033, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 3.19, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0357, |
|
"step": 2325 |
|
}, |
|
{ |
|
"epoch": 3.22, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0364, |
|
"step": 2350 |
|
}, |
|
{ |
|
"epoch": 3.26, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0425, |
|
"step": 2375 |
|
}, |
|
{ |
|
"epoch": 3.29, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0398, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 3.33, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0376, |
|
"step": 2425 |
|
}, |
|
{ |
|
"epoch": 3.36, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0413, |
|
"step": 2450 |
|
}, |
|
{ |
|
"epoch": 3.4, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0374, |
|
"step": 2475 |
|
}, |
|
{ |
|
"epoch": 3.43, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0383, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 3.46, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0381, |
|
"step": 2525 |
|
}, |
|
{ |
|
"epoch": 3.5, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0466, |
|
"step": 2550 |
|
}, |
|
{ |
|
"epoch": 3.53, |
|
"learning_rate": 1e-05, |
|
"loss": 0.04, |
|
"step": 2575 |
|
}, |
|
{ |
|
"epoch": 3.57, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0452, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 3.6, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0393, |
|
"step": 2625 |
|
}, |
|
{ |
|
"epoch": 3.64, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0461, |
|
"step": 2650 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0403, |
|
"step": 2675 |
|
}, |
|
{ |
|
"epoch": 3.7, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0489, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 3.74, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0418, |
|
"step": 2725 |
|
}, |
|
{ |
|
"epoch": 3.77, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0389, |
|
"step": 2750 |
|
}, |
|
{ |
|
"epoch": 3.81, |
|
"learning_rate": 1e-05, |
|
"loss": 0.041, |
|
"step": 2775 |
|
}, |
|
{ |
|
"epoch": 3.84, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0394, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 3.88, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0485, |
|
"step": 2825 |
|
}, |
|
{ |
|
"epoch": 3.91, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0416, |
|
"step": 2850 |
|
}, |
|
{ |
|
"epoch": 3.94, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0501, |
|
"step": 2875 |
|
}, |
|
{ |
|
"epoch": 3.98, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0393, |
|
"step": 2900 |
|
}, |
|
{ |
|
"epoch": 4.01, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0306, |
|
"step": 2925 |
|
}, |
|
{ |
|
"epoch": 4.05, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0265, |
|
"step": 2950 |
|
}, |
|
{ |
|
"epoch": 4.08, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0223, |
|
"step": 2975 |
|
}, |
|
{ |
|
"epoch": 4.12, |
|
"learning_rate": 1e-05, |
|
"loss": 0.0215, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 4.12, |
|
"eval_cer": 7.079429983827164, |
|
"eval_loss": 0.24198034405708313, |
|
"eval_runtime": 393.3959, |
|
"eval_samples_per_second": 1.932, |
|
"eval_steps_per_second": 0.122, |
|
"eval_wer": 17.98756798756799, |
|
"step": 3000 |
|
} |
|
], |
|
"max_steps": 4000, |
|
"num_train_epochs": 6, |
|
"total_flos": 1.384748184895488e+19, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|