epoch = 3.0
train_runtime = 19825.6129
train_samples_per_second = 1.118