train_results.json

{
    "epoch": 5.0,
    "init_mem_cpu_alloc_delta": -68857856,
    "init_mem_cpu_peaked_delta": 154255360,
    "init_mem_gpu_alloc_delta": 497570816,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 2823634944,
    "train_mem_cpu_peaked_delta": 164388864,
    "train_mem_gpu_alloc_delta": 1508407296,
    "train_mem_gpu_peaked_delta": 10039297024,
    "train_runtime": 1076.7664,
    "train_samples": 22648,
    "train_samples_per_second": 3.288
}