train_results.json

{
    "epoch": 3.0,
    "train_loss": 2.208091510499077,
    "train_runtime": 92.7531,
    "train_samples": 5000,
    "train_samples_per_second": 161.72,
    "train_steps_per_second": 2.555
}