epoch = 2.0
train_runtime = 14615.2615
train_samples_per_second = 2.138