epoch = 3.0
train_runtime = 113.2803
train_samples_per_second = 3.999