train_results.json

{
    "epoch": 3.0,
    "init_mem_cpu_alloc_delta": 1294196736,
    "init_mem_cpu_peaked_delta": 420376576,
    "init_mem_gpu_alloc_delta": 441293312,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 17265926144,
    "train_mem_cpu_peaked_delta": 799649792,
    "train_mem_gpu_alloc_delta": 1320941056,
    "train_mem_gpu_peaked_delta": 7582508544,
    "train_runtime": 388027.9373,
    "train_samples": 15140037,
    "train_samples_per_second": 1.829
}