epoch = 5.0
train_runtime = 3104.0027
train_samples_per_second = 0.459