all_results.json

{
    "epoch": 5.0,
    "eval_loss": 0.11634299904108047,
    "eval_runtime": 0.4543,
    "eval_samples_per_second": 1162.27,
    "eval_steps_per_second": 11.006,
    "max_memory_allocated (GB)": 94.43,
    "memory_allocated (GB)": 17.72,
    "total_memory_available (GB)": 94.61,
    "train_loss": 0.08891021746855515,
    "train_runtime": 341.352,
    "train_samples_per_second": 429.544,
    "train_steps_per_second": 1.141
}