epoch = 10.0
train_runtime = 15880.0798
train_samples_per_second = 7.871