epoch = 5.0
train_runtime = 15554.6344
train_samples_per_second = 1.709