train_results.json

{
    "epoch": 6.0,
    "train_loss": 1.235396307486075,
    "train_runtime": 303.3307,
    "train_samples": 5749,
    "train_samples_per_second": 947.646,
    "train_steps_per_second": 7.418
}