epoch = 0.98
train_runtime = 72359.2785
train_samples_per_second = 1.382