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 14
However, I don’t know if it’s possible to include the attention layers here as you do in your example.
model.fit(
    x=training['train']['input_ids'],
    y=training['labels_train'], 
    validation_data=(
        validation['valid']['input_ids'],
        validation['labels_test'],
    )
    ,batch_size = batch_size
    ,epochs = epochs
)