train_results.json

{
    "epoch": 5.0,
    "max_memory_allocated (GB)": 83.39,
    "memory_allocated (GB)": 19.37,
    "total_memory_available (GB)": 93.74,
    "train_loss": 0.07280745967741936,
    "train_runtime": 346.0342,
    "train_samples_per_second": 638.618,
    "train_steps_per_second": 2.022
}