{
    "epoch": 2.0,
    "max_memory_allocated (GB)": 91.87,
    "memory_allocated (GB)": 24.39,
    "total_memory_available (GB)": 94.62,
    "train_loss": 1.0469276263736165,
    "train_runtime": 6085.8982,
    "train_samples_per_second": 162.542,
    "train_steps_per_second": 0.317
}