all_results.json

{
    "epoch": 0.4,
    "learning_rate": 6.002269380684279e-06,
    "train_loss": 4.83620277834145,
    "train_runtime": 1683.6936,
    "train_samples": 2365187,
    "train_samples_per_second": 561.964,
    "train_steps_per_second": 1.098
}