epoch = 6.0
train_runtime = 106122.2527
train_samples_per_second = 11.818