{'train_runtime': 2297.443, 'train_samples_per_second': 4224.96, 'train_steps_per_second': 4.126, 'train_loss': 4.938072006611884, 'epoch': 1.0}