epoch = 3.0
train_runtime = 1176.2097
train_samples_per_second = 1.375