train_results.json

{
    "epoch": 3.0,
    "init_mem_cpu_alloc_delta": 1903816704,
    "init_mem_cpu_peaked_delta": 378310656,
    "init_mem_gpu_alloc_delta": 499890176,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 1347997696,
    "train_mem_cpu_peaked_delta": 317304832,
    "train_mem_gpu_alloc_delta": 1509584384,
    "train_mem_gpu_peaked_delta": 18134628864,
    "train_runtime": 11284.7294,
    "train_samples": 392702,
    "train_samples_per_second": 0.816
}