all_results.json

{
    "epoch": 1.0,
    "train_loss": 0.1562451661368947,
    "train_runtime": 3209.2418,
    "train_samples": 51894,
    "train_samples_per_second": 16.17,
    "train_steps_per_second": 0.126
}