{
    "epoch": 1.0,
    "max_memory_allocated (GB)": 42.87,
    "memory_allocated (GB)": 35.52,
    "total_memory_available (GB)": 94.62,
    "train_loss": 1.7192556858062744,
    "train_runtime": 30.1519,
    "train_samples_per_second": -0.696,
    "train_steps_per_second": -0.066
}