train_results.json

{
    "epoch": 30.0,
    "init_mem_cpu_alloc_delta": 10140165,
    "init_mem_cpu_peaked_delta": 18306,
    "init_mem_gpu_alloc_delta": 1261939712,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 1543641,
    "train_mem_cpu_peaked_delta": 478977252,
    "train_mem_gpu_alloc_delta": 3779933184,
    "train_mem_gpu_peaked_delta": 7771228672,
    "train_runtime": 43862.7197,
    "train_samples": 11106,
    "train_samples_per_second": 0.238
}