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