all_results.json

{
    "epoch": 2.0,
    "train_loss": 0.030777697094727392,
    "train_runtime": 228453.6417,
    "train_samples": 1116594,
    "train_samples_per_second": 9.775,
    "train_steps_per_second": 9.775
}