epoch = 5.0
train_runtime = 8275.4395
train_samples_per_second = 1.217