{ "best_metric": 35.03184713375796, "best_model_checkpoint": "./whisper-small-trained/checkpoint-3000", "epoch": 22.099447513812155, "global_step": 4000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.14, "learning_rate": 4.2000000000000006e-07, "loss": 2.2821, "step": 25 }, { "epoch": 0.28, "learning_rate": 9.200000000000001e-07, "loss": 1.8517, "step": 50 }, { "epoch": 0.41, "learning_rate": 1.42e-06, "loss": 1.5837, "step": 75 }, { "epoch": 0.55, "learning_rate": 1.9000000000000002e-06, "loss": 1.5654, "step": 100 }, { "epoch": 0.69, "learning_rate": 2.4000000000000003e-06, "loss": 1.4586, "step": 125 }, { "epoch": 0.83, "learning_rate": 2.9e-06, "loss": 1.022, "step": 150 }, { "epoch": 0.97, "learning_rate": 3.4000000000000005e-06, "loss": 1.0495, "step": 175 }, { "epoch": 1.1, "learning_rate": 3.900000000000001e-06, "loss": 0.7301, "step": 200 }, { "epoch": 1.24, "learning_rate": 4.4e-06, "loss": 0.6992, "step": 225 }, { "epoch": 1.38, "learning_rate": 4.9000000000000005e-06, "loss": 0.6385, "step": 250 }, { "epoch": 1.52, "learning_rate": 5.400000000000001e-06, "loss": 0.7749, "step": 275 }, { "epoch": 1.66, "learning_rate": 5.9e-06, "loss": 0.5811, "step": 300 }, { "epoch": 1.8, "learning_rate": 6.4000000000000006e-06, "loss": 0.5882, "step": 325 }, { "epoch": 1.93, "learning_rate": 6.9e-06, "loss": 0.5089, "step": 350 }, { "epoch": 2.07, "learning_rate": 7.4e-06, "loss": 0.345, "step": 375 }, { "epoch": 2.21, "learning_rate": 7.9e-06, "loss": 0.1977, "step": 400 }, { "epoch": 2.35, "learning_rate": 8.400000000000001e-06, "loss": 0.3083, "step": 425 }, { "epoch": 2.49, "learning_rate": 8.900000000000001e-06, "loss": 0.2266, "step": 450 }, { "epoch": 2.62, "learning_rate": 9.4e-06, "loss": 0.2451, "step": 475 }, { "epoch": 2.76, "learning_rate": 9.9e-06, "loss": 0.3034, "step": 500 }, { "epoch": 2.9, "learning_rate": 9.942857142857145e-06, "loss": 0.2351, "step": 525 }, { "epoch": 3.04, "learning_rate": 9.871428571428572e-06, "loss": 0.186, "step": 550 }, { "epoch": 3.18, "learning_rate": 9.800000000000001e-06, "loss": 0.132, "step": 575 }, { "epoch": 3.31, "learning_rate": 9.72857142857143e-06, "loss": 0.0805, "step": 600 }, { "epoch": 3.45, "learning_rate": 9.657142857142859e-06, "loss": 0.1079, "step": 625 }, { "epoch": 3.59, "learning_rate": 9.585714285714286e-06, "loss": 0.1255, "step": 650 }, { "epoch": 3.73, "learning_rate": 9.514285714285715e-06, "loss": 0.1197, "step": 675 }, { "epoch": 3.87, "learning_rate": 9.442857142857144e-06, "loss": 0.1106, "step": 700 }, { "epoch": 4.01, "learning_rate": 9.371428571428572e-06, "loss": 0.1102, "step": 725 }, { "epoch": 4.14, "learning_rate": 9.3e-06, "loss": 0.0513, "step": 750 }, { "epoch": 4.28, "learning_rate": 9.22857142857143e-06, "loss": 0.0353, "step": 775 }, { "epoch": 4.42, "learning_rate": 9.157142857142857e-06, "loss": 0.0585, "step": 800 }, { "epoch": 4.56, "learning_rate": 9.085714285714286e-06, "loss": 0.0626, "step": 825 }, { "epoch": 4.7, "learning_rate": 9.014285714285715e-06, "loss": 0.0688, "step": 850 }, { "epoch": 4.83, "learning_rate": 8.942857142857142e-06, "loss": 0.0707, "step": 875 }, { "epoch": 4.97, "learning_rate": 8.871428571428571e-06, "loss": 0.0454, "step": 900 }, { "epoch": 5.11, "learning_rate": 8.8e-06, "loss": 0.0343, "step": 925 }, { "epoch": 5.25, "learning_rate": 8.72857142857143e-06, "loss": 0.0302, "step": 950 }, { "epoch": 5.39, "learning_rate": 8.657142857142858e-06, "loss": 0.0412, "step": 975 }, { "epoch": 5.52, "learning_rate": 8.585714285714286e-06, "loss": 0.0442, "step": 1000 }, { "epoch": 5.52, "eval_loss": 0.8577815890312195, "eval_runtime": 13.8198, "eval_samples_per_second": 3.401, "eval_steps_per_second": 0.434, "eval_wer": 35.29723991507431, "step": 1000 }, { "epoch": 5.66, "learning_rate": 8.514285714285715e-06, "loss": 0.0274, "step": 1025 }, { "epoch": 5.8, "learning_rate": 8.442857142857144e-06, "loss": 0.0302, "step": 1050 }, { "epoch": 5.94, "learning_rate": 8.371428571428573e-06, "loss": 0.0322, "step": 1075 }, { "epoch": 6.08, "learning_rate": 8.3e-06, "loss": 0.0225, "step": 1100 }, { "epoch": 6.22, "learning_rate": 8.22857142857143e-06, "loss": 0.0186, "step": 1125 }, { "epoch": 6.35, "learning_rate": 8.157142857142858e-06, "loss": 0.0099, "step": 1150 }, { "epoch": 6.49, "learning_rate": 8.085714285714287e-06, "loss": 0.0147, "step": 1175 }, { "epoch": 6.63, "learning_rate": 8.014285714285715e-06, "loss": 0.0171, "step": 1200 }, { "epoch": 6.77, "learning_rate": 7.942857142857144e-06, "loss": 0.0179, "step": 1225 }, { "epoch": 6.91, "learning_rate": 7.871428571428573e-06, "loss": 0.0243, "step": 1250 }, { "epoch": 7.04, "learning_rate": 7.800000000000002e-06, "loss": 0.0078, "step": 1275 }, { "epoch": 7.18, "learning_rate": 7.72857142857143e-06, "loss": 0.0109, "step": 1300 }, { "epoch": 7.32, "learning_rate": 7.657142857142858e-06, "loss": 0.0125, "step": 1325 }, { "epoch": 7.46, "learning_rate": 7.585714285714286e-06, "loss": 0.013, "step": 1350 }, { "epoch": 7.6, "learning_rate": 7.514285714285715e-06, "loss": 0.0069, "step": 1375 }, { "epoch": 7.73, "learning_rate": 7.442857142857144e-06, "loss": 0.0143, "step": 1400 }, { "epoch": 7.87, "learning_rate": 7.371428571428571e-06, "loss": 0.0109, "step": 1425 }, { "epoch": 8.01, "learning_rate": 7.3e-06, "loss": 0.0085, "step": 1450 }, { "epoch": 8.15, "learning_rate": 7.2285714285714294e-06, "loss": 0.0035, "step": 1475 }, { "epoch": 8.29, "learning_rate": 7.1571428571428584e-06, "loss": 0.0141, "step": 1500 }, { "epoch": 8.43, "learning_rate": 7.085714285714286e-06, "loss": 0.0059, "step": 1525 }, { "epoch": 8.56, "learning_rate": 7.014285714285715e-06, "loss": 0.0073, "step": 1550 }, { "epoch": 8.7, "learning_rate": 6.942857142857144e-06, "loss": 0.0058, "step": 1575 }, { "epoch": 8.84, "learning_rate": 6.871428571428572e-06, "loss": 0.0045, "step": 1600 }, { "epoch": 8.98, "learning_rate": 6.800000000000001e-06, "loss": 0.0026, "step": 1625 }, { "epoch": 9.12, "learning_rate": 6.728571428571429e-06, "loss": 0.0008, "step": 1650 }, { "epoch": 9.25, "learning_rate": 6.657142857142857e-06, "loss": 0.0039, "step": 1675 }, { "epoch": 9.39, "learning_rate": 6.585714285714286e-06, "loss": 0.0051, "step": 1700 }, { "epoch": 9.53, "learning_rate": 6.514285714285715e-06, "loss": 0.0049, "step": 1725 }, { "epoch": 9.67, "learning_rate": 6.442857142857143e-06, "loss": 0.0087, "step": 1750 }, { "epoch": 9.81, "learning_rate": 6.371428571428572e-06, "loss": 0.0023, "step": 1775 }, { "epoch": 9.94, "learning_rate": 6.300000000000001e-06, "loss": 0.0077, "step": 1800 }, { "epoch": 10.08, "learning_rate": 6.22857142857143e-06, "loss": 0.0064, "step": 1825 }, { "epoch": 10.22, "learning_rate": 6.157142857142858e-06, "loss": 0.0004, "step": 1850 }, { "epoch": 10.36, "learning_rate": 6.085714285714286e-06, "loss": 0.002, "step": 1875 }, { "epoch": 10.5, "learning_rate": 6.014285714285715e-06, "loss": 0.0047, "step": 1900 }, { "epoch": 10.64, "learning_rate": 5.942857142857143e-06, "loss": 0.0042, "step": 1925 }, { "epoch": 10.77, "learning_rate": 5.871428571428572e-06, "loss": 0.0028, "step": 1950 }, { "epoch": 10.91, "learning_rate": 5.8e-06, "loss": 0.0018, "step": 1975 }, { "epoch": 11.05, "learning_rate": 5.7285714285714285e-06, "loss": 0.0043, "step": 2000 }, { "epoch": 11.05, "eval_loss": 0.9808950424194336, "eval_runtime": 11.6591, "eval_samples_per_second": 4.031, "eval_steps_per_second": 0.515, "eval_wer": 37.95116772823779, "step": 2000 }, { "epoch": 11.19, "learning_rate": 5.6571428571428576e-06, "loss": 0.0003, "step": 2025 }, { "epoch": 11.33, "learning_rate": 5.5857142857142866e-06, "loss": 0.0003, "step": 2050 }, { "epoch": 11.46, "learning_rate": 5.514285714285714e-06, "loss": 0.0018, "step": 2075 }, { "epoch": 11.6, "learning_rate": 5.442857142857143e-06, "loss": 0.005, "step": 2100 }, { "epoch": 11.74, "learning_rate": 5.371428571428572e-06, "loss": 0.0005, "step": 2125 }, { "epoch": 11.88, "learning_rate": 5.300000000000001e-06, "loss": 0.0034, "step": 2150 }, { "epoch": 12.02, "learning_rate": 5.22857142857143e-06, "loss": 0.0003, "step": 2175 }, { "epoch": 12.15, "learning_rate": 5.157142857142857e-06, "loss": 0.0003, "step": 2200 }, { "epoch": 12.29, "learning_rate": 5.085714285714286e-06, "loss": 0.0004, "step": 2225 }, { "epoch": 12.43, "learning_rate": 5.014285714285715e-06, "loss": 0.0003, "step": 2250 }, { "epoch": 12.57, "learning_rate": 4.9428571428571435e-06, "loss": 0.0004, "step": 2275 }, { "epoch": 12.71, "learning_rate": 4.871428571428572e-06, "loss": 0.0014, "step": 2300 }, { "epoch": 12.85, "learning_rate": 4.800000000000001e-06, "loss": 0.002, "step": 2325 }, { "epoch": 12.98, "learning_rate": 4.728571428571429e-06, "loss": 0.0003, "step": 2350 }, { "epoch": 13.12, "learning_rate": 4.657142857142857e-06, "loss": 0.0002, "step": 2375 }, { "epoch": 13.26, "learning_rate": 4.585714285714286e-06, "loss": 0.0039, "step": 2400 }, { "epoch": 13.4, "learning_rate": 4.514285714285714e-06, "loss": 0.0002, "step": 2425 }, { "epoch": 13.54, "learning_rate": 4.442857142857143e-06, "loss": 0.0003, "step": 2450 }, { "epoch": 13.67, "learning_rate": 4.371428571428572e-06, "loss": 0.0002, "step": 2475 }, { "epoch": 13.81, "learning_rate": 4.3e-06, "loss": 0.0002, "step": 2500 }, { "epoch": 13.95, "learning_rate": 4.228571428571429e-06, "loss": 0.0002, "step": 2525 }, { "epoch": 14.09, "learning_rate": 4.1571428571428575e-06, "loss": 0.0003, "step": 2550 }, { "epoch": 14.23, "learning_rate": 4.0857142857142865e-06, "loss": 0.0002, "step": 2575 }, { "epoch": 14.36, "learning_rate": 4.014285714285715e-06, "loss": 0.0002, "step": 2600 }, { "epoch": 14.5, "learning_rate": 3.942857142857143e-06, "loss": 0.0002, "step": 2625 }, { "epoch": 14.64, "learning_rate": 3.871428571428572e-06, "loss": 0.0002, "step": 2650 }, { "epoch": 14.78, "learning_rate": 3.8000000000000005e-06, "loss": 0.0003, "step": 2675 }, { "epoch": 14.92, "learning_rate": 3.7285714285714286e-06, "loss": 0.0002, "step": 2700 }, { "epoch": 15.06, "learning_rate": 3.6571428571428576e-06, "loss": 0.0002, "step": 2725 }, { "epoch": 15.19, "learning_rate": 3.5857142857142862e-06, "loss": 0.0002, "step": 2750 }, { "epoch": 15.33, "learning_rate": 3.5142857142857144e-06, "loss": 0.0024, "step": 2775 }, { "epoch": 15.47, "learning_rate": 3.4428571428571434e-06, "loss": 0.0002, "step": 2800 }, { "epoch": 15.61, "learning_rate": 3.3714285714285716e-06, "loss": 0.0002, "step": 2825 }, { "epoch": 15.75, "learning_rate": 3.3000000000000006e-06, "loss": 0.0002, "step": 2850 }, { "epoch": 15.88, "learning_rate": 3.2285714285714288e-06, "loss": 0.0002, "step": 2875 }, { "epoch": 16.02, "learning_rate": 3.1571428571428573e-06, "loss": 0.0002, "step": 2900 }, { "epoch": 16.16, "learning_rate": 3.085714285714286e-06, "loss": 0.0001, "step": 2925 }, { "epoch": 16.3, "learning_rate": 3.0142857142857145e-06, "loss": 0.0001, "step": 2950 }, { "epoch": 16.44, "learning_rate": 2.9428571428571427e-06, "loss": 0.0001, "step": 2975 }, { "epoch": 16.57, "learning_rate": 2.8714285714285717e-06, "loss": 0.0002, "step": 3000 }, { "epoch": 16.57, "eval_loss": 1.0312094688415527, "eval_runtime": 10.1748, "eval_samples_per_second": 4.619, "eval_steps_per_second": 0.59, "eval_wer": 35.03184713375796, "step": 3000 }, { "epoch": 16.71, "learning_rate": 2.8000000000000003e-06, "loss": 0.0001, "step": 3025 }, { "epoch": 16.85, "learning_rate": 2.728571428571429e-06, "loss": 0.0002, "step": 3050 }, { "epoch": 16.99, "learning_rate": 2.6571428571428575e-06, "loss": 0.0001, "step": 3075 }, { "epoch": 17.13, "learning_rate": 2.5857142857142856e-06, "loss": 0.0001, "step": 3100 }, { "epoch": 17.27, "learning_rate": 2.5142857142857147e-06, "loss": 0.0002, "step": 3125 }, { "epoch": 17.4, "learning_rate": 2.442857142857143e-06, "loss": 0.0001, "step": 3150 }, { "epoch": 17.54, "learning_rate": 2.371428571428572e-06, "loss": 0.0001, "step": 3175 }, { "epoch": 17.68, "learning_rate": 2.3000000000000004e-06, "loss": 0.0001, "step": 3200 }, { "epoch": 17.82, "learning_rate": 2.228571428571429e-06, "loss": 0.0001, "step": 3225 }, { "epoch": 17.96, "learning_rate": 2.157142857142857e-06, "loss": 0.0001, "step": 3250 }, { "epoch": 18.09, "learning_rate": 2.0857142857142858e-06, "loss": 0.0001, "step": 3275 }, { "epoch": 18.23, "learning_rate": 2.0142857142857144e-06, "loss": 0.0001, "step": 3300 }, { "epoch": 18.37, "learning_rate": 1.942857142857143e-06, "loss": 0.0001, "step": 3325 }, { "epoch": 18.51, "learning_rate": 1.8714285714285715e-06, "loss": 0.0001, "step": 3350 }, { "epoch": 18.65, "learning_rate": 1.8000000000000001e-06, "loss": 0.0001, "step": 3375 }, { "epoch": 18.78, "learning_rate": 1.7285714285714287e-06, "loss": 0.0001, "step": 3400 }, { "epoch": 18.92, "learning_rate": 1.657142857142857e-06, "loss": 0.0001, "step": 3425 }, { "epoch": 19.06, "learning_rate": 1.5857142857142857e-06, "loss": 0.0001, "step": 3450 }, { "epoch": 19.2, "learning_rate": 1.5142857142857145e-06, "loss": 0.0001, "step": 3475 }, { "epoch": 19.34, "learning_rate": 1.442857142857143e-06, "loss": 0.0001, "step": 3500 }, { "epoch": 19.48, "learning_rate": 1.3714285714285717e-06, "loss": 0.0001, "step": 3525 }, { "epoch": 19.61, "learning_rate": 1.3e-06, "loss": 0.0001, "step": 3550 }, { "epoch": 19.75, "learning_rate": 1.2285714285714286e-06, "loss": 0.0001, "step": 3575 }, { "epoch": 19.89, "learning_rate": 1.1571428571428572e-06, "loss": 0.0001, "step": 3600 }, { "epoch": 20.03, "learning_rate": 1.0857142857142858e-06, "loss": 0.0001, "step": 3625 }, { "epoch": 20.17, "learning_rate": 1.0142857142857144e-06, "loss": 0.0001, "step": 3650 }, { "epoch": 20.3, "learning_rate": 9.42857142857143e-07, "loss": 0.0001, "step": 3675 }, { "epoch": 20.44, "learning_rate": 8.714285714285716e-07, "loss": 0.0001, "step": 3700 }, { "epoch": 20.58, "learning_rate": 8.000000000000001e-07, "loss": 0.0001, "step": 3725 }, { "epoch": 20.72, "learning_rate": 7.285714285714287e-07, "loss": 0.0001, "step": 3750 }, { "epoch": 20.86, "learning_rate": 6.571428571428571e-07, "loss": 0.0001, "step": 3775 }, { "epoch": 20.99, "learning_rate": 5.857142857142857e-07, "loss": 0.0001, "step": 3800 }, { "epoch": 21.13, "learning_rate": 5.142857142857143e-07, "loss": 0.0001, "step": 3825 }, { "epoch": 21.27, "learning_rate": 4.4285714285714286e-07, "loss": 0.0001, "step": 3850 }, { "epoch": 21.41, "learning_rate": 3.7142857142857145e-07, "loss": 0.0001, "step": 3875 }, { "epoch": 21.55, "learning_rate": 3.0000000000000004e-07, "loss": 0.0001, "step": 3900 }, { "epoch": 21.69, "learning_rate": 2.285714285714286e-07, "loss": 0.0001, "step": 3925 }, { "epoch": 21.82, "learning_rate": 1.5714285714285717e-07, "loss": 0.0001, "step": 3950 }, { "epoch": 21.96, "learning_rate": 8.571428571428573e-08, "loss": 0.0001, "step": 3975 }, { "epoch": 22.1, "learning_rate": 1.4285714285714288e-08, "loss": 0.0001, "step": 4000 }, { "epoch": 22.1, "eval_loss": 1.0490719079971313, "eval_runtime": 10.3052, "eval_samples_per_second": 4.561, "eval_steps_per_second": 0.582, "eval_wer": 35.138004246284495, "step": 4000 } ], "max_steps": 4000, "num_train_epochs": 23, "total_flos": 1.15434160128e+18, "trial_name": null, "trial_params": null }