all_results.json

{
    "epoch": 3.0,
    "max_memory_allocated (GB)": 9.66,
    "memory_allocated (GB)": 8.43,
    "total_memory_available (GB)": 94.62,
    "train_loss": 3.116956232882094,
    "train_runtime": 155.3604,
    "train_samples": 2318,
    "train_samples_per_second": 46.845,
    "train_steps_per_second": 11.721
}