{
    "epoch": 5.0,
    "init_mem_cpu_alloc_delta": 2353512448,
    "init_mem_cpu_peaked_delta": 261173248,
    "init_mem_gpu_alloc_delta": 268956160,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 3074768896,
    "train_mem_cpu_peaked_delta": 7049216,
    "train_mem_gpu_alloc_delta": 814749184,
    "train_mem_gpu_peaked_delta": 9148695040,
    "train_runtime": 9244.0598,
    "train_samples": 392702,
    "train_samples_per_second": 1.659
}