all_results.json

{
    "epoch": 0.4,
    "learning_rate": 6.004432654828931e-06,
    "train_loss": 4.836088716499729,
    "train_runtime": 1690.4281,
    "train_samples": 2364278,
    "train_samples_per_second": 559.423,
    "train_steps_per_second": 1.093
}