epoch = 1.0
train_loss = 2.1594684836158007
train_runtime = 6223.7531
train_samples_per_second = 223.702
train_steps_per_second = 1.748