I’m not sure if it’s possible to print metrics with that object, but with model.fit it automatically displays the metrics as in TensorFlow.
Check out the course page
However, I don’t know if it’s possible to include the attention layers here as you do in your example.