epoch = 5.0
train_runtime = 883.0149
train_samples_per_second = 16.483