epoch = 50.0
init_mem_cpu_alloc_delta = 335619
init_mem_cpu_peaked_delta = 18306
init_mem_gpu_alloc_delta = 334744576
init_mem_gpu_peaked_delta = 0
train_mem_cpu_alloc_delta = 703692
train_mem_cpu_peaked_delta = 309631449
train_mem_gpu_alloc_delta = 996869632
train_mem_gpu_peaked_delta = 9629578240
train_runtime = 2224.787
train_samples_per_second = 3.326