train_results.json

{
    "epoch": 3.0,
    "init_mem_cpu_alloc_delta": 1912971264,
    "init_mem_cpu_peaked_delta": 448593920,
    "init_mem_gpu_alloc_delta": 433779200,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 1157033984,
    "train_mem_cpu_peaked_delta": 277667840,
    "train_mem_gpu_alloc_delta": 1315372032,
    "train_mem_gpu_peaked_delta": 18135096320,
    "train_runtime": 15561.9859,
    "train_samples": 550152,
    "train_samples_per_second": 0.829
}