epoch = 6.0
train_loss = 1.717211879325718
train_runtime = 1445.1612
train_samples_per_second = 93.037
train_steps_per_second = 2.91