epoch = 10.0
train_runtime = 7427.3988
train_samples_per_second = 3.925