all_results.json

{
    "epoch": 20.0,
    "train_loss": 1.1193655486238139,
    "train_runtime": 133767.8538,
    "train_samples_per_second": 3.998,
    "train_steps_per_second": 0.5
}