{
    "epoch": 3.0,
    "max_memory_allocated (GB)": 74.17,
    "memory_allocated (GB)": 59.44,
    "total_memory_available (GB)": 94.62,
    "train_loss": 1.3051700514580127,
    "train_runtime": 1023.0938,
    "train_samples_per_second": 18.938,
    "train_steps_per_second": 1.185
}