Get Embeddings from BERT after fine tuning

Hey,

I am following the following script to fine tune the BERT base model on my dataset.
https://gluon-nlp.mxnet.io/examples/sentence_embedding/bert.html

Objective: To get sentence embeddings from fine tuned BERT to use in sentence similarity task

Problem : I have fine tuned using the script given here:
https://gluon-nlp.mxnet.io/examples/sentence_embedding/bert.html

I wanted to know the code of how to get the sentence embeddings from this code at the end after fine tuning ?

Thanks,