epoch = 0.98
train_runtime = 106961.6151
train_samples_per_second = 0.935