train_results.json

{
    "epoch": 2.0,
    "train_loss": 1.51298133963187,
    "train_runtime": 13015.1099,
    "train_samples": 132768,
    "train_samples_per_second": 20.402,
    "train_steps_per_second": 1.7
}