epoch = 5.0
train_runtime = 938.0621
train_samples_per_second = 2.409