train_results.json

{
    "epoch": 7.0,
    "init_mem_cpu_alloc_delta": 3210498048,
    "init_mem_cpu_peaked_delta": 319852544,
    "init_mem_gpu_alloc_delta": 439069184,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 86511616,
    "train_mem_cpu_peaked_delta": 187301888,
    "train_mem_gpu_alloc_delta": 1378340352,
    "train_mem_gpu_peaked_delta": 3560736768,
    "train_runtime": 851.9812,
    "train_samples": 5749,
    "train_samples_per_second": 1.479
}