all_results.json

{
    "epoch": 5.0,
    "train_loss": 3.531251077090993,
    "train_runtime": 47.3239,
    "train_samples": 1671,
    "train_samples_per_second": 176.549,
    "train_steps_per_second": 3.592
}