epoch = 1.0
train_runtime = 31399.0337
train_samples_per_second = 6.066