all_results.json

{
    "epoch": 5.0,
    "eval_loss": 0.12152472138404846,
    "eval_runtime": 3.7753,
    "eval_samples_per_second": 141.222,
    "eval_steps_per_second": 8.826,
    "max_memory_allocated (GB)": 60.52,
    "memory_allocated (GB)": 51.27,
    "total_flos": 3.0598946525952e+16,
    "total_memory_available (GB)": 94.62,
    "train_loss": 0.06085505417415074,
    "train_runtime": 1020.8061,
    "train_samples_per_second": 55.51,
    "train_steps_per_second": 1.389
}