train_results.json

{
    "epoch": 2.0,
    "train_loss": 0.3625062153889583,
    "train_runtime": 97159.7175,
    "train_samples_per_second": 5.892,
    "train_steps_per_second": 0.023
}