{'train_runtime': 2015.1004, 'train_samples_per_second': 4816.933, 'train_steps_per_second': 4.704, 'train_loss': 1.5626397716848157, 'epoch': 1.0}