epoch = 0.98
train_runtime = 76223.6887
train_samples_per_second = 1.312