{
    "epoch": 2.0,
    "max_memory_allocated (GB)": 30.34,
    "memory_allocated (GB)": 19.36,
    "total_memory_available (GB)": 94.62,
    "train_loss": 2.0642123500715774,
    "train_runtime": 549.193,
    "train_samples_per_second": 97.186,
    "train_steps_per_second": 1.517
}