all_results.json

{
    "epoch": 20.0,
    "train_loss": 0.3965469028917141,
    "train_runtime": 374.4336,
    "train_samples_per_second": 1.709,
    "train_steps_per_second": 0.427
}