train_results.json

{
    "epoch": 20.41,
    "init_mem_cpu_alloc_delta": 275763200,
    "init_mem_cpu_peaked_delta": 0,
    "init_mem_gpu_alloc_delta": 1481701376,
    "init_mem_gpu_peaked_delta": 0,
    "total_flos": 4484754685255680.0,
    "train_mem_cpu_alloc_delta": 403214336,
    "train_mem_cpu_peaked_delta": 1180577792,
    "train_mem_gpu_alloc_delta": 4476536832,
    "train_mem_gpu_peaked_delta": 4057436672,
    "train_runtime": 3032.248,
    "train_samples": 194,
    "train_samples_per_second": 0.33
}