epoch = 10.0 train_loss = 1.6471780081788416 train_runtime = 2443.9537 train_samples_per_second = 91.692 train_steps_per_second = 2.868