epoch = 10.0
train_runtime = 57584.0925
train_samples_per_second = 2.171