train_results.json

{
    "epoch": 1.0,
    "max_memory_allocated (GB)": 66.62,
    "memory_allocated (GB)": 47.23,
    "total_memory_available (GB)": 94.62,
    "train_loss": 0.9383976405317133,
    "train_runtime": 33.3025,
    "train_samples_per_second": 23.852,
    "train_steps_per_second": 1.645
}