train_results.json

{
    "epoch": 3.0,
    "init_mem_cpu_alloc_delta": 2093309952,
    "init_mem_cpu_peaked_delta": 265920512,
    "init_mem_gpu_alloc_delta": 329781248,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 729513984,
    "train_mem_cpu_peaked_delta": 483684352,
    "train_mem_gpu_alloc_delta": 1020158976,
    "train_mem_gpu_peaked_delta": 2316007936,
    "train_runtime": 10484.4187,
    "train_samples": 392702,
    "train_samples_per_second": 3.511
}