{
    "before_init_mem_cpu": 929583104,
    "before_init_mem_gpu": 0,
    "epoch": 81.0,
    "init_mem_cpu_alloc_delta": 2783653888,
    "init_mem_cpu_peaked_delta": 411418624,
    "init_mem_gpu_alloc_delta": 497994240,
    "init_mem_gpu_peaked_delta": 512,
    "train_loss": 0.019815802188779586,
    "train_mem_cpu_alloc_delta": 8908660736,
    "train_mem_cpu_peaked_delta": 387145728,
    "train_mem_gpu_alloc_delta": 2006552576,
    "train_mem_gpu_peaked_delta": 15468659712,
    "train_runtime": 22056.5171,
    "train_samples": 3976976,
    "train_samples_per_second": 14604.983,
    "train_steps_per_second": 1.781
}