| { | |
| "best_metric": 13.861446892066024, | |
| "best_model_checkpoint": "./checkpoint-2000", | |
| "epoch": 27.932960893854748, | |
| "global_step": 5000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 2.4000000000000003e-06, | |
| "loss": 0.2213, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.56, | |
| "learning_rate": 4.9000000000000005e-06, | |
| "loss": 0.0366, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "learning_rate": 7.4e-06, | |
| "loss": 0.0253, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.12, | |
| "learning_rate": 9.9e-06, | |
| "loss": 0.0235, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.4, | |
| "learning_rate": 9.9e-06, | |
| "loss": 0.0204, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.68, | |
| "learning_rate": 9.795833333333334e-06, | |
| "loss": 0.0173, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.96, | |
| "learning_rate": 9.691666666666667e-06, | |
| "loss": 0.0196, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.23, | |
| "learning_rate": 9.587500000000001e-06, | |
| "loss": 0.0109, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.51, | |
| "learning_rate": 9.483333333333335e-06, | |
| "loss": 0.009, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 2.79, | |
| "learning_rate": 9.379166666666667e-06, | |
| "loss": 0.0107, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.07, | |
| "learning_rate": 9.275e-06, | |
| "loss": 0.0097, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 3.35, | |
| "learning_rate": 9.170833333333334e-06, | |
| "loss": 0.0044, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.63, | |
| "learning_rate": 9.066666666666667e-06, | |
| "loss": 0.0049, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 3.91, | |
| "learning_rate": 8.962500000000001e-06, | |
| "loss": 0.0061, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 4.19, | |
| "learning_rate": 8.858333333333335e-06, | |
| "loss": 0.0042, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 4.47, | |
| "learning_rate": 8.754166666666667e-06, | |
| "loss": 0.0028, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 4.75, | |
| "learning_rate": 8.65e-06, | |
| "loss": 0.0036, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 5.03, | |
| "learning_rate": 8.545833333333334e-06, | |
| "loss": 0.0036, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 5.31, | |
| "learning_rate": 8.441666666666667e-06, | |
| "loss": 0.0019, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 5.59, | |
| "learning_rate": 8.337500000000001e-06, | |
| "loss": 0.0022, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 5.59, | |
| "eval_loss": 0.24051959812641144, | |
| "eval_runtime": 738.2945, | |
| "eval_samples_per_second": 9.403, | |
| "eval_steps_per_second": 0.294, | |
| "eval_wer": 13.999509112106523, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 5.87, | |
| "learning_rate": 8.233333333333335e-06, | |
| "loss": 0.0022, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 6.15, | |
| "learning_rate": 8.129166666666667e-06, | |
| "loss": 0.0026, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 6.42, | |
| "learning_rate": 8.025e-06, | |
| "loss": 0.0014, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 6.7, | |
| "learning_rate": 7.920833333333334e-06, | |
| "loss": 0.0016, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 6.98, | |
| "learning_rate": 7.816666666666667e-06, | |
| "loss": 0.0017, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 7.26, | |
| "learning_rate": 7.712500000000001e-06, | |
| "loss": 0.0013, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 7.54, | |
| "learning_rate": 7.608333333333334e-06, | |
| "loss": 0.0012, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 7.82, | |
| "learning_rate": 7.504166666666667e-06, | |
| "loss": 0.0016, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 8.1, | |
| "learning_rate": 7.4e-06, | |
| "loss": 0.0013, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 8.38, | |
| "learning_rate": 7.295833333333334e-06, | |
| "loss": 0.0009, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 8.66, | |
| "learning_rate": 7.191666666666667e-06, | |
| "loss": 0.0013, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 8.94, | |
| "learning_rate": 7.0875e-06, | |
| "loss": 0.0014, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 9.22, | |
| "learning_rate": 6.983333333333334e-06, | |
| "loss": 0.001, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "learning_rate": 6.879166666666667e-06, | |
| "loss": 0.0009, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 9.78, | |
| "learning_rate": 6.775e-06, | |
| "loss": 0.0008, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 10.06, | |
| "learning_rate": 6.670833333333334e-06, | |
| "loss": 0.0008, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 10.34, | |
| "learning_rate": 6.566666666666667e-06, | |
| "loss": 0.0009, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 10.61, | |
| "learning_rate": 6.4625e-06, | |
| "loss": 0.0007, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 10.89, | |
| "learning_rate": 6.358333333333334e-06, | |
| "loss": 0.0008, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 11.17, | |
| "learning_rate": 6.254166666666667e-06, | |
| "loss": 0.0005, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 11.17, | |
| "eval_loss": 0.2584499418735504, | |
| "eval_runtime": 741.1067, | |
| "eval_samples_per_second": 9.367, | |
| "eval_steps_per_second": 0.293, | |
| "eval_wer": 13.861446892066024, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 11.45, | |
| "learning_rate": 6.15e-06, | |
| "loss": 0.0007, | |
| "step": 2050 | |
| }, | |
| { | |
| "epoch": 11.73, | |
| "learning_rate": 6.045833333333334e-06, | |
| "loss": 0.0006, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 12.01, | |
| "learning_rate": 5.941666666666667e-06, | |
| "loss": 0.0009, | |
| "step": 2150 | |
| }, | |
| { | |
| "epoch": 12.29, | |
| "learning_rate": 5.8375000000000004e-06, | |
| "loss": 0.0005, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 12.57, | |
| "learning_rate": 5.733333333333334e-06, | |
| "loss": 0.0004, | |
| "step": 2250 | |
| }, | |
| { | |
| "epoch": 12.85, | |
| "learning_rate": 5.629166666666667e-06, | |
| "loss": 0.0006, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 13.13, | |
| "learning_rate": 5.5250000000000005e-06, | |
| "loss": 0.0004, | |
| "step": 2350 | |
| }, | |
| { | |
| "epoch": 13.41, | |
| "learning_rate": 5.420833333333334e-06, | |
| "loss": 0.0003, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 13.69, | |
| "learning_rate": 5.316666666666667e-06, | |
| "loss": 0.0004, | |
| "step": 2450 | |
| }, | |
| { | |
| "epoch": 13.97, | |
| "learning_rate": 5.2125000000000005e-06, | |
| "loss": 0.0006, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 14.25, | |
| "learning_rate": 5.108333333333334e-06, | |
| "loss": 0.0003, | |
| "step": 2550 | |
| }, | |
| { | |
| "epoch": 14.53, | |
| "learning_rate": 5.004166666666667e-06, | |
| "loss": 0.0002, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 14.8, | |
| "learning_rate": 4.9000000000000005e-06, | |
| "loss": 0.0002, | |
| "step": 2650 | |
| }, | |
| { | |
| "epoch": 15.08, | |
| "learning_rate": 4.795833333333333e-06, | |
| "loss": 0.0004, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 15.36, | |
| "learning_rate": 4.691666666666667e-06, | |
| "loss": 0.0002, | |
| "step": 2750 | |
| }, | |
| { | |
| "epoch": 15.64, | |
| "learning_rate": 4.5875000000000005e-06, | |
| "loss": 0.0002, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 15.92, | |
| "learning_rate": 4.483333333333333e-06, | |
| "loss": 0.0003, | |
| "step": 2850 | |
| }, | |
| { | |
| "epoch": 16.2, | |
| "learning_rate": 4.379166666666667e-06, | |
| "loss": 0.0003, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 16.48, | |
| "learning_rate": 4.2750000000000006e-06, | |
| "loss": 0.0003, | |
| "step": 2950 | |
| }, | |
| { | |
| "epoch": 16.76, | |
| "learning_rate": 4.170833333333333e-06, | |
| "loss": 0.0002, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 16.76, | |
| "eval_loss": 0.26832228899002075, | |
| "eval_runtime": 755.3783, | |
| "eval_samples_per_second": 9.19, | |
| "eval_steps_per_second": 0.287, | |
| "eval_wer": 16.559796281524207, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 17.04, | |
| "learning_rate": 4.066666666666667e-06, | |
| "loss": 0.0001, | |
| "step": 3050 | |
| }, | |
| { | |
| "epoch": 17.32, | |
| "learning_rate": 3.962500000000001e-06, | |
| "loss": 0.0003, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 17.6, | |
| "learning_rate": 3.858333333333333e-06, | |
| "loss": 0.0001, | |
| "step": 3150 | |
| }, | |
| { | |
| "epoch": 17.88, | |
| "learning_rate": 3.754166666666667e-06, | |
| "loss": 0.0002, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 18.16, | |
| "learning_rate": 3.65e-06, | |
| "loss": 0.0001, | |
| "step": 3250 | |
| }, | |
| { | |
| "epoch": 18.44, | |
| "learning_rate": 3.5458333333333334e-06, | |
| "loss": 0.0001, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 18.72, | |
| "learning_rate": 3.441666666666667e-06, | |
| "loss": 0.0002, | |
| "step": 3350 | |
| }, | |
| { | |
| "epoch": 18.99, | |
| "learning_rate": 3.3375000000000002e-06, | |
| "loss": 0.0001, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 19.27, | |
| "learning_rate": 3.2333333333333334e-06, | |
| "loss": 0.0001, | |
| "step": 3450 | |
| }, | |
| { | |
| "epoch": 19.55, | |
| "learning_rate": 3.129166666666667e-06, | |
| "loss": 0.0001, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 19.83, | |
| "learning_rate": 3.0250000000000003e-06, | |
| "loss": 0.0001, | |
| "step": 3550 | |
| }, | |
| { | |
| "epoch": 20.11, | |
| "learning_rate": 2.9208333333333335e-06, | |
| "loss": 0.0, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 20.39, | |
| "learning_rate": 2.816666666666667e-06, | |
| "loss": 0.0, | |
| "step": 3650 | |
| }, | |
| { | |
| "epoch": 20.67, | |
| "learning_rate": 2.7125000000000003e-06, | |
| "loss": 0.0001, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 20.95, | |
| "learning_rate": 2.608333333333333e-06, | |
| "loss": 0.0, | |
| "step": 3750 | |
| }, | |
| { | |
| "epoch": 21.23, | |
| "learning_rate": 2.504166666666667e-06, | |
| "loss": 0.0001, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 21.51, | |
| "learning_rate": 2.4000000000000003e-06, | |
| "loss": 0.0, | |
| "step": 3850 | |
| }, | |
| { | |
| "epoch": 21.79, | |
| "learning_rate": 2.2958333333333335e-06, | |
| "loss": 0.0001, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 22.07, | |
| "learning_rate": 2.191666666666667e-06, | |
| "loss": 0.0, | |
| "step": 3950 | |
| }, | |
| { | |
| "epoch": 22.35, | |
| "learning_rate": 2.0875e-06, | |
| "loss": 0.0, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 22.35, | |
| "eval_loss": 0.2907346189022064, | |
| "eval_runtime": 746.9633, | |
| "eval_samples_per_second": 9.294, | |
| "eval_steps_per_second": 0.291, | |
| "eval_wer": 15.038043811744492, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 22.63, | |
| "learning_rate": 1.9833333333333335e-06, | |
| "loss": 0.0, | |
| "step": 4050 | |
| }, | |
| { | |
| "epoch": 22.91, | |
| "learning_rate": 1.879166666666667e-06, | |
| "loss": 0.0, | |
| "step": 4100 | |
| }, | |
| { | |
| "epoch": 23.18, | |
| "learning_rate": 1.7750000000000002e-06, | |
| "loss": 0.0, | |
| "step": 4150 | |
| }, | |
| { | |
| "epoch": 23.46, | |
| "learning_rate": 1.6708333333333336e-06, | |
| "loss": 0.0, | |
| "step": 4200 | |
| }, | |
| { | |
| "epoch": 23.74, | |
| "learning_rate": 1.566666666666667e-06, | |
| "loss": 0.0, | |
| "step": 4250 | |
| }, | |
| { | |
| "epoch": 24.02, | |
| "learning_rate": 1.4625e-06, | |
| "loss": 0.0, | |
| "step": 4300 | |
| }, | |
| { | |
| "epoch": 24.3, | |
| "learning_rate": 1.3583333333333334e-06, | |
| "loss": 0.0, | |
| "step": 4350 | |
| }, | |
| { | |
| "epoch": 24.58, | |
| "learning_rate": 1.2541666666666668e-06, | |
| "loss": 0.0, | |
| "step": 4400 | |
| }, | |
| { | |
| "epoch": 24.86, | |
| "learning_rate": 1.1500000000000002e-06, | |
| "loss": 0.0, | |
| "step": 4450 | |
| }, | |
| { | |
| "epoch": 25.14, | |
| "learning_rate": 1.0458333333333334e-06, | |
| "loss": 0.0, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 25.42, | |
| "learning_rate": 9.416666666666667e-07, | |
| "loss": 0.0, | |
| "step": 4550 | |
| }, | |
| { | |
| "epoch": 25.7, | |
| "learning_rate": 8.375000000000001e-07, | |
| "loss": 0.0, | |
| "step": 4600 | |
| }, | |
| { | |
| "epoch": 25.98, | |
| "learning_rate": 7.333333333333334e-07, | |
| "loss": 0.0, | |
| "step": 4650 | |
| }, | |
| { | |
| "epoch": 26.26, | |
| "learning_rate": 6.291666666666666e-07, | |
| "loss": 0.0, | |
| "step": 4700 | |
| }, | |
| { | |
| "epoch": 26.54, | |
| "learning_rate": 5.250000000000001e-07, | |
| "loss": 0.0, | |
| "step": 4750 | |
| }, | |
| { | |
| "epoch": 26.82, | |
| "learning_rate": 4.2083333333333336e-07, | |
| "loss": 0.0, | |
| "step": 4800 | |
| }, | |
| { | |
| "epoch": 27.09, | |
| "learning_rate": 3.166666666666667e-07, | |
| "loss": 0.0, | |
| "step": 4850 | |
| }, | |
| { | |
| "epoch": 27.37, | |
| "learning_rate": 2.1250000000000003e-07, | |
| "loss": 0.0, | |
| "step": 4900 | |
| }, | |
| { | |
| "epoch": 27.65, | |
| "learning_rate": 1.0833333333333335e-07, | |
| "loss": 0.0, | |
| "step": 4950 | |
| }, | |
| { | |
| "epoch": 27.93, | |
| "learning_rate": 4.1666666666666676e-09, | |
| "loss": 0.0, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 27.93, | |
| "eval_loss": 0.3084936738014221, | |
| "eval_runtime": 745.3939, | |
| "eval_samples_per_second": 9.313, | |
| "eval_steps_per_second": 0.291, | |
| "eval_wer": 14.203534392833037, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 27.93, | |
| "step": 5000, | |
| "total_flos": 4.611132960473088e+19, | |
| "train_loss": 0.004727415198316157, | |
| "train_runtime": 12262.7443, | |
| "train_samples_per_second": 13.048, | |
| "train_steps_per_second": 0.408 | |
| } | |
| ], | |
| "max_steps": 5000, | |
| "num_train_epochs": 28, | |
| "total_flos": 4.611132960473088e+19, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |