train_results.json

{
    "epoch": 20.0,
    "init_mem_cpu_alloc_delta": 2547835,
    "init_mem_cpu_peaked_delta": 38772,
    "train_mem_cpu_alloc_delta": 1063085,
    "train_mem_cpu_peaked_delta": 209273878,
    "train_runtime": 264850.0877,
    "train_samples": 438452,
    "train_samples_per_second": 1.035
}