all_results.json

{
    "epoch": 5.0,
    "train_loss": 0.8455495816299968,
    "train_runtime": 715.3867,
    "train_samples": 9481,
    "train_samples_per_second": 66.265,
    "train_steps_per_second": 2.216
}