epoch = 0.98
train_runtime = 75126.7844
train_samples_per_second = 1.331