epoch = 5.0
train_runtime = 19.5675
train_samples_per_second = 9.71