train_results.json

{
    "epoch": 1.0,
    "train_loss": 1.4067512356623162,
    "train_runtime": 5393.9428,
    "train_samples": 801861,
    "train_samples_per_second": 148.66,
    "train_steps_per_second": 4.646
}