epoch = 10.0
train_runtime = 35714.6581
train_samples_per_second = 3.5