all_results.json

{
    "epoch": 2.0,
    "train_loss": 3.5304283500368654,
    "train_runtime": 15323.5402,
    "train_samples": 131854,
    "train_samples_per_second": 17.209,
    "train_steps_per_second": 2.868
}