all_results.json

{
    "epoch": 20.0,
    "train_loss": 0.3553026157890757,
    "train_runtime": 2958.1152,
    "train_samples_per_second": 4.057,
    "train_steps_per_second": 1.014
}