| { | |
| "best_metric": 50.85625, | |
| "best_model_checkpoint": "./whisper-small-shona/checkpoint-1200", | |
| "epoch": 166.64, | |
| "global_step": 2000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 2.08, | |
| "learning_rate": 4.800000000000001e-07, | |
| "loss": 3.009, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 4.16, | |
| "learning_rate": 9.600000000000001e-07, | |
| "loss": 2.4459, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 6.24, | |
| "learning_rate": 1.46e-06, | |
| "loss": 1.8015, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 8.32, | |
| "learning_rate": 1.9600000000000003e-06, | |
| "loss": 1.2656, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 10.4, | |
| "learning_rate": 2.46e-06, | |
| "loss": 0.9172, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 12.48, | |
| "learning_rate": 2.96e-06, | |
| "loss": 0.6728, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 14.56, | |
| "learning_rate": 3.46e-06, | |
| "loss": 0.4792, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 16.64, | |
| "learning_rate": 3.96e-06, | |
| "loss": 0.32, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 18.72, | |
| "learning_rate": 4.4600000000000005e-06, | |
| "loss": 0.1975, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 20.8, | |
| "learning_rate": 4.960000000000001e-06, | |
| "loss": 0.1101, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 22.88, | |
| "learning_rate": 5.460000000000001e-06, | |
| "loss": 0.0537, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 24.96, | |
| "learning_rate": 5.9600000000000005e-06, | |
| "loss": 0.0256, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 27.08, | |
| "learning_rate": 6.460000000000001e-06, | |
| "loss": 0.014, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 29.16, | |
| "learning_rate": 6.96e-06, | |
| "loss": 0.0092, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 31.24, | |
| "learning_rate": 7.4600000000000006e-06, | |
| "loss": 0.007, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 33.32, | |
| "learning_rate": 7.960000000000002e-06, | |
| "loss": 0.0054, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 33.32, | |
| "eval_loss": 0.9825747013092041, | |
| "eval_runtime": 245.236, | |
| "eval_samples_per_second": 3.772, | |
| "eval_steps_per_second": 0.061, | |
| "eval_wer": 51.668749999999996, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 35.4, | |
| "learning_rate": 8.46e-06, | |
| "loss": 0.0043, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 37.48, | |
| "learning_rate": 8.96e-06, | |
| "loss": 0.0036, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 39.56, | |
| "learning_rate": 9.460000000000001e-06, | |
| "loss": 0.003, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 41.64, | |
| "learning_rate": 9.960000000000001e-06, | |
| "loss": 0.0026, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 43.72, | |
| "learning_rate": 9.846666666666668e-06, | |
| "loss": 0.0022, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 45.8, | |
| "learning_rate": 9.68e-06, | |
| "loss": 0.0019, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 47.88, | |
| "learning_rate": 9.513333333333334e-06, | |
| "loss": 0.0017, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 49.96, | |
| "learning_rate": 9.346666666666666e-06, | |
| "loss": 0.0016, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 52.08, | |
| "learning_rate": 9.180000000000002e-06, | |
| "loss": 0.0015, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 54.16, | |
| "learning_rate": 9.013333333333334e-06, | |
| "loss": 0.0013, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 56.24, | |
| "learning_rate": 8.846666666666668e-06, | |
| "loss": 0.0012, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 58.32, | |
| "learning_rate": 8.68e-06, | |
| "loss": 0.0011, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 60.4, | |
| "learning_rate": 8.513333333333335e-06, | |
| "loss": 0.001, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 62.48, | |
| "learning_rate": 8.346666666666668e-06, | |
| "loss": 0.001, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 64.56, | |
| "learning_rate": 8.18e-06, | |
| "loss": 0.0009, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 66.64, | |
| "learning_rate": 8.013333333333333e-06, | |
| "loss": 0.0009, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 66.64, | |
| "eval_loss": 1.0774297714233398, | |
| "eval_runtime": 228.2657, | |
| "eval_samples_per_second": 4.052, | |
| "eval_steps_per_second": 0.066, | |
| "eval_wer": 50.90625, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 68.72, | |
| "learning_rate": 7.846666666666667e-06, | |
| "loss": 0.0008, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 70.8, | |
| "learning_rate": 7.680000000000001e-06, | |
| "loss": 0.0008, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 72.88, | |
| "learning_rate": 7.513333333333334e-06, | |
| "loss": 0.0007, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 74.96, | |
| "learning_rate": 7.346666666666668e-06, | |
| "loss": 0.0007, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 77.08, | |
| "learning_rate": 7.180000000000001e-06, | |
| "loss": 0.0007, | |
| "step": 925 | |
| }, | |
| { | |
| "epoch": 79.16, | |
| "learning_rate": 7.0133333333333345e-06, | |
| "loss": 0.0007, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 81.24, | |
| "learning_rate": 6.846666666666667e-06, | |
| "loss": 0.0006, | |
| "step": 975 | |
| }, | |
| { | |
| "epoch": 83.32, | |
| "learning_rate": 6.680000000000001e-06, | |
| "loss": 0.0006, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 85.4, | |
| "learning_rate": 6.513333333333333e-06, | |
| "loss": 0.0006, | |
| "step": 1025 | |
| }, | |
| { | |
| "epoch": 87.48, | |
| "learning_rate": 6.346666666666668e-06, | |
| "loss": 0.0006, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 89.56, | |
| "learning_rate": 6.18e-06, | |
| "loss": 0.0005, | |
| "step": 1075 | |
| }, | |
| { | |
| "epoch": 91.64, | |
| "learning_rate": 6.013333333333335e-06, | |
| "loss": 0.0005, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 93.72, | |
| "learning_rate": 5.846666666666667e-06, | |
| "loss": 0.0005, | |
| "step": 1125 | |
| }, | |
| { | |
| "epoch": 95.8, | |
| "learning_rate": 5.68e-06, | |
| "loss": 0.0005, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 97.88, | |
| "learning_rate": 5.513333333333334e-06, | |
| "loss": 0.0005, | |
| "step": 1175 | |
| }, | |
| { | |
| "epoch": 99.96, | |
| "learning_rate": 5.346666666666667e-06, | |
| "loss": 0.0005, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 99.96, | |
| "eval_loss": 1.117426872253418, | |
| "eval_runtime": 245.3452, | |
| "eval_samples_per_second": 3.77, | |
| "eval_steps_per_second": 0.061, | |
| "eval_wer": 50.85625, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 102.08, | |
| "learning_rate": 5.18e-06, | |
| "loss": 0.0005, | |
| "step": 1225 | |
| }, | |
| { | |
| "epoch": 104.16, | |
| "learning_rate": 5.013333333333333e-06, | |
| "loss": 0.0004, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 106.24, | |
| "learning_rate": 4.846666666666667e-06, | |
| "loss": 0.0004, | |
| "step": 1275 | |
| }, | |
| { | |
| "epoch": 108.32, | |
| "learning_rate": 4.680000000000001e-06, | |
| "loss": 0.0004, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 110.4, | |
| "learning_rate": 4.513333333333333e-06, | |
| "loss": 0.0004, | |
| "step": 1325 | |
| }, | |
| { | |
| "epoch": 112.48, | |
| "learning_rate": 4.346666666666667e-06, | |
| "loss": 0.0004, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 114.56, | |
| "learning_rate": 4.18e-06, | |
| "loss": 0.0004, | |
| "step": 1375 | |
| }, | |
| { | |
| "epoch": 116.64, | |
| "learning_rate": 4.013333333333334e-06, | |
| "loss": 0.0004, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 118.72, | |
| "learning_rate": 3.8466666666666665e-06, | |
| "loss": 0.0004, | |
| "step": 1425 | |
| }, | |
| { | |
| "epoch": 120.8, | |
| "learning_rate": 3.6800000000000003e-06, | |
| "loss": 0.0004, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 122.88, | |
| "learning_rate": 3.5133333333333337e-06, | |
| "loss": 0.0004, | |
| "step": 1475 | |
| }, | |
| { | |
| "epoch": 124.96, | |
| "learning_rate": 3.346666666666667e-06, | |
| "loss": 0.0004, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 127.08, | |
| "learning_rate": 3.1800000000000005e-06, | |
| "loss": 0.0004, | |
| "step": 1525 | |
| }, | |
| { | |
| "epoch": 129.16, | |
| "learning_rate": 3.013333333333334e-06, | |
| "loss": 0.0004, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 131.24, | |
| "learning_rate": 2.8466666666666672e-06, | |
| "loss": 0.0003, | |
| "step": 1575 | |
| }, | |
| { | |
| "epoch": 133.32, | |
| "learning_rate": 2.68e-06, | |
| "loss": 0.0003, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 133.32, | |
| "eval_loss": 1.138814926147461, | |
| "eval_runtime": 244.5662, | |
| "eval_samples_per_second": 3.782, | |
| "eval_steps_per_second": 0.061, | |
| "eval_wer": 50.875, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 135.4, | |
| "learning_rate": 2.5133333333333336e-06, | |
| "loss": 0.0003, | |
| "step": 1625 | |
| }, | |
| { | |
| "epoch": 137.48, | |
| "learning_rate": 2.346666666666667e-06, | |
| "loss": 0.0003, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 139.56, | |
| "learning_rate": 2.1800000000000003e-06, | |
| "loss": 0.0003, | |
| "step": 1675 | |
| }, | |
| { | |
| "epoch": 141.64, | |
| "learning_rate": 2.0133333333333337e-06, | |
| "loss": 0.0003, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 143.72, | |
| "learning_rate": 1.8466666666666668e-06, | |
| "loss": 0.0003, | |
| "step": 1725 | |
| }, | |
| { | |
| "epoch": 145.8, | |
| "learning_rate": 1.6800000000000002e-06, | |
| "loss": 0.0003, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 147.88, | |
| "learning_rate": 1.5133333333333334e-06, | |
| "loss": 0.0003, | |
| "step": 1775 | |
| }, | |
| { | |
| "epoch": 149.96, | |
| "learning_rate": 1.3466666666666668e-06, | |
| "loss": 0.0003, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 152.08, | |
| "learning_rate": 1.1800000000000001e-06, | |
| "loss": 0.0003, | |
| "step": 1825 | |
| }, | |
| { | |
| "epoch": 154.16, | |
| "learning_rate": 1.0133333333333333e-06, | |
| "loss": 0.0003, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 156.24, | |
| "learning_rate": 8.466666666666668e-07, | |
| "loss": 0.0003, | |
| "step": 1875 | |
| }, | |
| { | |
| "epoch": 158.32, | |
| "learning_rate": 6.800000000000001e-07, | |
| "loss": 0.0003, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 160.4, | |
| "learning_rate": 5.133333333333334e-07, | |
| "loss": 0.0003, | |
| "step": 1925 | |
| }, | |
| { | |
| "epoch": 162.48, | |
| "learning_rate": 3.466666666666667e-07, | |
| "loss": 0.0003, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 164.56, | |
| "learning_rate": 1.8e-07, | |
| "loss": 0.0003, | |
| "step": 1975 | |
| }, | |
| { | |
| "epoch": 166.64, | |
| "learning_rate": 1.3333333333333334e-08, | |
| "loss": 0.0003, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 166.64, | |
| "eval_loss": 1.1461412906646729, | |
| "eval_runtime": 244.933, | |
| "eval_samples_per_second": 3.777, | |
| "eval_steps_per_second": 0.061, | |
| "eval_wer": 50.925, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 166.64, | |
| "step": 2000, | |
| "total_flos": 3.770541405645465e+19, | |
| "train_loss": 0.1423063586682547, | |
| "train_runtime": 4512.4735, | |
| "train_samples_per_second": 28.366, | |
| "train_steps_per_second": 0.443 | |
| } | |
| ], | |
| "max_steps": 2000, | |
| "num_train_epochs": 167, | |
| "total_flos": 3.770541405645465e+19, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |