all_results.json

{
    "epoch": 20.0,
    "train_loss": 1.952947931640795,
    "train_runtime": 385301.7422,
    "train_samples": 551465,
    "train_samples_per_second": 28.625,
    "train_steps_per_second": 1.789
}