epoch = 3.0
train_runtime = 488.8595
train_samples_per_second = 6.462