{
  "best_metric": 3.3522434244455903,
  "best_model_checkpoint": "./whisper-small/checkpoint-2000",
  "epoch": 6.472491909385114,
  "eval_steps": 10,
  "global_step": 2000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.08,
      "grad_norm": 81.5032730102539,
      "learning_rate": 4.2000000000000006e-07,
      "loss": 2.9603,
      "step": 25
    },
    {
      "epoch": 0.16,
      "grad_norm": 17.11170768737793,
      "learning_rate": 9.200000000000001e-07,
      "loss": 2.144,
      "step": 50
    },
    {
      "epoch": 0.24,
      "grad_norm": 11.967795372009277,
      "learning_rate": 1.42e-06,
      "loss": 1.3446,
      "step": 75
    },
    {
      "epoch": 0.32,
      "grad_norm": 8.235530853271484,
      "learning_rate": 1.9200000000000003e-06,
      "loss": 0.7532,
      "step": 100
    },
    {
      "epoch": 0.4,
      "grad_norm": 6.933337688446045,
      "learning_rate": 2.42e-06,
      "loss": 0.5681,
      "step": 125
    },
    {
      "epoch": 0.49,
      "grad_norm": 8.942669868469238,
      "learning_rate": 2.92e-06,
      "loss": 0.496,
      "step": 150
    },
    {
      "epoch": 0.57,
      "grad_norm": 11.463906288146973,
      "learning_rate": 3.4200000000000007e-06,
      "loss": 0.4111,
      "step": 175
    },
    {
      "epoch": 0.65,
      "grad_norm": 7.532162666320801,
      "learning_rate": 3.920000000000001e-06,
      "loss": 0.3332,
      "step": 200
    },
    {
      "epoch": 0.73,
      "grad_norm": 7.5595197677612305,
      "learning_rate": 4.42e-06,
      "loss": 0.2728,
      "step": 225
    },
    {
      "epoch": 0.81,
      "grad_norm": 4.813773155212402,
      "learning_rate": 4.92e-06,
      "loss": 0.1087,
      "step": 250
    },
    {
      "epoch": 0.89,
      "grad_norm": 3.7373056411743164,
      "learning_rate": 5.420000000000001e-06,
      "loss": 0.0653,
      "step": 275
    },
    {
      "epoch": 0.97,
      "grad_norm": 4.375819683074951,
      "learning_rate": 5.92e-06,
      "loss": 0.059,
      "step": 300
    },
    {
      "epoch": 1.05,
      "grad_norm": 1.661054015159607,
      "learning_rate": 6.42e-06,
      "loss": 0.0381,
      "step": 325
    },
    {
      "epoch": 1.13,
      "grad_norm": 2.9067795276641846,
      "learning_rate": 6.92e-06,
      "loss": 0.0327,
      "step": 350
    },
    {
      "epoch": 1.21,
      "grad_norm": 4.853701114654541,
      "learning_rate": 7.420000000000001e-06,
      "loss": 0.0295,
      "step": 375
    },
    {
      "epoch": 1.29,
      "grad_norm": 1.736449122428894,
      "learning_rate": 7.92e-06,
      "loss": 0.0311,
      "step": 400
    },
    {
      "epoch": 1.38,
      "grad_norm": 3.658984899520874,
      "learning_rate": 8.42e-06,
      "loss": 0.0296,
      "step": 425
    },
    {
      "epoch": 1.46,
      "grad_norm": 2.040172576904297,
      "learning_rate": 8.920000000000001e-06,
      "loss": 0.0221,
      "step": 450
    },
    {
      "epoch": 1.54,
      "grad_norm": 2.4981894493103027,
      "learning_rate": 9.42e-06,
      "loss": 0.0251,
      "step": 475
    },
    {
      "epoch": 1.62,
      "grad_norm": 2.639390230178833,
      "learning_rate": 9.920000000000002e-06,
      "loss": 0.0261,
      "step": 500
    },
    {
      "epoch": 1.7,
      "grad_norm": 2.526102066040039,
      "learning_rate": 9.940000000000001e-06,
      "loss": 0.0253,
      "step": 525
    },
    {
      "epoch": 1.78,
      "grad_norm": 1.3722740411758423,
      "learning_rate": 9.86857142857143e-06,
      "loss": 0.0203,
      "step": 550
    },
    {
      "epoch": 1.86,
      "grad_norm": 2.3701043128967285,
      "learning_rate": 9.797142857142858e-06,
      "loss": 0.0225,
      "step": 575
    },
    {
      "epoch": 1.94,
      "grad_norm": 2.8724324703216553,
      "learning_rate": 9.725714285714287e-06,
      "loss": 0.0201,
      "step": 600
    },
    {
      "epoch": 2.02,
      "grad_norm": 1.0449714660644531,
      "learning_rate": 9.654285714285716e-06,
      "loss": 0.0149,
      "step": 625
    },
    {
      "epoch": 2.1,
      "grad_norm": 5.038928031921387,
      "learning_rate": 9.582857142857143e-06,
      "loss": 0.0104,
      "step": 650
    },
    {
      "epoch": 2.18,
      "grad_norm": 1.6107451915740967,
      "learning_rate": 9.511428571428572e-06,
      "loss": 0.0086,
      "step": 675
    },
    {
      "epoch": 2.27,
      "grad_norm": 1.3224350214004517,
      "learning_rate": 9.440000000000001e-06,
      "loss": 0.006,
      "step": 700
    },
    {
      "epoch": 2.35,
      "grad_norm": 0.14248408377170563,
      "learning_rate": 9.368571428571428e-06,
      "loss": 0.0059,
      "step": 725
    },
    {
      "epoch": 2.43,
      "grad_norm": 2.3801496028900146,
      "learning_rate": 9.297142857142857e-06,
      "loss": 0.0056,
      "step": 750
    },
    {
      "epoch": 2.51,
      "grad_norm": 1.156813621520996,
      "learning_rate": 9.225714285714286e-06,
      "loss": 0.0067,
      "step": 775
    },
    {
      "epoch": 2.59,
      "grad_norm": 0.9941128492355347,
      "learning_rate": 9.154285714285715e-06,
      "loss": 0.0076,
      "step": 800
    },
    {
      "epoch": 2.67,
      "grad_norm": 1.4023518562316895,
      "learning_rate": 9.082857142857143e-06,
      "loss": 0.006,
      "step": 825
    },
    {
      "epoch": 2.75,
      "grad_norm": 1.1814390420913696,
      "learning_rate": 9.011428571428572e-06,
      "loss": 0.0053,
      "step": 850
    },
    {
      "epoch": 2.83,
      "grad_norm": 1.8819189071655273,
      "learning_rate": 8.94e-06,
      "loss": 0.008,
      "step": 875
    },
    {
      "epoch": 2.91,
      "grad_norm": 0.9754526019096375,
      "learning_rate": 8.86857142857143e-06,
      "loss": 0.0047,
      "step": 900
    },
    {
      "epoch": 2.99,
      "grad_norm": 0.40376096963882446,
      "learning_rate": 8.797142857142857e-06,
      "loss": 0.0062,
      "step": 925
    },
    {
      "epoch": 3.07,
      "grad_norm": 0.07103849202394485,
      "learning_rate": 8.725714285714286e-06,
      "loss": 0.0028,
      "step": 950
    },
    {
      "epoch": 3.16,
      "grad_norm": 0.5621446967124939,
      "learning_rate": 8.654285714285715e-06,
      "loss": 0.0025,
      "step": 975
    },
    {
      "epoch": 3.24,
      "grad_norm": 0.506003201007843,
      "learning_rate": 8.582857142857144e-06,
      "loss": 0.0026,
      "step": 1000
    },
    {
      "epoch": 3.24,
      "eval_loss": 0.020510252565145493,
      "eval_runtime": 791.6891,
      "eval_samples_per_second": 1.557,
      "eval_steps_per_second": 0.196,
      "eval_wer": 4.486848891181021,
      "step": 1000
    },
    {
      "epoch": 3.32,
      "grad_norm": 0.723780632019043,
      "learning_rate": 8.511428571428571e-06,
      "loss": 0.003,
      "step": 1025
    },
    {
      "epoch": 3.4,
      "grad_norm": 0.02721921168267727,
      "learning_rate": 8.44e-06,
      "loss": 0.0028,
      "step": 1050
    },
    {
      "epoch": 3.48,
      "grad_norm": 1.0563775300979614,
      "learning_rate": 8.36857142857143e-06,
      "loss": 0.002,
      "step": 1075
    },
    {
      "epoch": 3.56,
      "grad_norm": 1.54115629196167,
      "learning_rate": 8.297142857142859e-06,
      "loss": 0.0027,
      "step": 1100
    },
    {
      "epoch": 3.64,
      "grad_norm": 0.017975807189941406,
      "learning_rate": 8.225714285714288e-06,
      "loss": 0.0016,
      "step": 1125
    },
    {
      "epoch": 3.72,
      "grad_norm": 0.15264229476451874,
      "learning_rate": 8.154285714285715e-06,
      "loss": 0.0034,
      "step": 1150
    },
    {
      "epoch": 3.8,
      "grad_norm": 0.655139148235321,
      "learning_rate": 8.082857142857144e-06,
      "loss": 0.0025,
      "step": 1175
    },
    {
      "epoch": 3.88,
      "grad_norm": 0.9618725776672363,
      "learning_rate": 8.011428571428573e-06,
      "loss": 0.0017,
      "step": 1200
    },
    {
      "epoch": 3.96,
      "grad_norm": 3.206550121307373,
      "learning_rate": 7.94e-06,
      "loss": 0.0031,
      "step": 1225
    },
    {
      "epoch": 4.05,
      "grad_norm": 0.41130420565605164,
      "learning_rate": 7.86857142857143e-06,
      "loss": 0.0012,
      "step": 1250
    },
    {
      "epoch": 4.13,
      "grad_norm": 0.0929947942495346,
      "learning_rate": 7.797142857142858e-06,
      "loss": 0.0012,
      "step": 1275
    },
    {
      "epoch": 4.21,
      "grad_norm": 0.050773363560438156,
      "learning_rate": 7.725714285714286e-06,
      "loss": 0.0005,
      "step": 1300
    },
    {
      "epoch": 4.29,
      "grad_norm": 0.0827890932559967,
      "learning_rate": 7.654285714285715e-06,
      "loss": 0.0005,
      "step": 1325
    },
    {
      "epoch": 4.37,
      "grad_norm": 0.02210867777466774,
      "learning_rate": 7.5828571428571444e-06,
      "loss": 0.0007,
      "step": 1350
    },
    {
      "epoch": 4.45,
      "grad_norm": 0.014925210736691952,
      "learning_rate": 7.511428571428572e-06,
      "loss": 0.0012,
      "step": 1375
    },
    {
      "epoch": 4.53,
      "grad_norm": 0.027211977168917656,
      "learning_rate": 7.440000000000001e-06,
      "loss": 0.0009,
      "step": 1400
    },
    {
      "epoch": 4.61,
      "grad_norm": 1.4061143398284912,
      "learning_rate": 7.36857142857143e-06,
      "loss": 0.0008,
      "step": 1425
    },
    {
      "epoch": 4.69,
      "grad_norm": 0.1997688263654709,
      "learning_rate": 7.297142857142858e-06,
      "loss": 0.0013,
      "step": 1450
    },
    {
      "epoch": 4.77,
      "grad_norm": 0.011363201774656773,
      "learning_rate": 7.225714285714286e-06,
      "loss": 0.0005,
      "step": 1475
    },
    {
      "epoch": 4.85,
      "grad_norm": 0.020546328276395798,
      "learning_rate": 7.154285714285715e-06,
      "loss": 0.0004,
      "step": 1500
    },
    {
      "epoch": 4.94,
      "grad_norm": 0.026502510532736778,
      "learning_rate": 7.082857142857143e-06,
      "loss": 0.0007,
      "step": 1525
    },
    {
      "epoch": 5.02,
      "grad_norm": 0.1575097292661667,
      "learning_rate": 7.011428571428572e-06,
      "loss": 0.0009,
      "step": 1550
    },
    {
      "epoch": 5.1,
      "grad_norm": 0.89250248670578,
      "learning_rate": 6.9400000000000005e-06,
      "loss": 0.001,
      "step": 1575
    },
    {
      "epoch": 5.18,
      "grad_norm": 0.5113543272018433,
      "learning_rate": 6.868571428571429e-06,
      "loss": 0.0003,
      "step": 1600
    },
    {
      "epoch": 5.26,
      "grad_norm": 0.05453397333621979,
      "learning_rate": 6.797142857142858e-06,
      "loss": 0.0003,
      "step": 1625
    },
    {
      "epoch": 5.34,
      "grad_norm": 0.01590396836400032,
      "learning_rate": 6.725714285714287e-06,
      "loss": 0.0002,
      "step": 1650
    },
    {
      "epoch": 5.42,
      "grad_norm": 0.00851709395647049,
      "learning_rate": 6.654285714285716e-06,
      "loss": 0.0005,
      "step": 1675
    },
    {
      "epoch": 5.5,
      "grad_norm": 0.014185091480612755,
      "learning_rate": 6.582857142857143e-06,
      "loss": 0.0004,
      "step": 1700
    },
    {
      "epoch": 5.58,
      "grad_norm": 0.017834780737757683,
      "learning_rate": 6.511428571428572e-06,
      "loss": 0.0003,
      "step": 1725
    },
    {
      "epoch": 5.66,
      "grad_norm": 0.017684083431959152,
      "learning_rate": 6.440000000000001e-06,
      "loss": 0.0003,
      "step": 1750
    },
    {
      "epoch": 5.74,
      "grad_norm": 0.009825997054576874,
      "learning_rate": 6.368571428571429e-06,
      "loss": 0.0001,
      "step": 1775
    },
    {
      "epoch": 5.83,
      "grad_norm": 0.08621440827846527,
      "learning_rate": 6.297142857142857e-06,
      "loss": 0.0004,
      "step": 1800
    },
    {
      "epoch": 5.91,
      "grad_norm": 0.019367387518286705,
      "learning_rate": 6.225714285714286e-06,
      "loss": 0.0007,
      "step": 1825
    },
    {
      "epoch": 5.99,
      "grad_norm": 0.04589274898171425,
      "learning_rate": 6.1542857142857145e-06,
      "loss": 0.0003,
      "step": 1850
    },
    {
      "epoch": 6.07,
      "grad_norm": 0.010185680352151394,
      "learning_rate": 6.0828571428571435e-06,
      "loss": 0.0002,
      "step": 1875
    },
    {
      "epoch": 6.15,
      "grad_norm": 0.004478767979890108,
      "learning_rate": 6.011428571428572e-06,
      "loss": 0.0002,
      "step": 1900
    },
    {
      "epoch": 6.23,
      "grad_norm": 0.03858327120542526,
      "learning_rate": 5.94e-06,
      "loss": 0.0001,
      "step": 1925
    },
    {
      "epoch": 6.31,
      "grad_norm": 0.008098349906504154,
      "learning_rate": 5.868571428571429e-06,
      "loss": 0.0002,
      "step": 1950
    },
    {
      "epoch": 6.39,
      "grad_norm": 0.0035556757356971502,
      "learning_rate": 5.797142857142858e-06,
      "loss": 0.0001,
      "step": 1975
    },
    {
      "epoch": 6.47,
      "grad_norm": 0.00917895045131445,
      "learning_rate": 5.725714285714287e-06,
      "loss": 0.0003,
      "step": 2000
    },
    {
      "epoch": 6.47,
      "eval_loss": 0.01798020303249359,
      "eval_runtime": 807.9427,
      "eval_samples_per_second": 1.526,
      "eval_steps_per_second": 0.192,
      "eval_wer": 3.3522434244455903,
      "step": 2000
    }
  ],
  "logging_steps": 25,
  "max_steps": 4000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 13,
  "save_steps": 10,
  "total_flos": 9.21395466141696e+18,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}