train_results.json

{
    "epoch": 0.4,
    "learning_rate": 5.993616284105673e-06,
    "train_loss": 4.707593353469222,
    "train_runtime": 1724.2288,
    "train_samples": 2369810,
    "train_samples_per_second": 549.941,
    "train_steps_per_second": 1.074
}