Get Embeddings from BERT after fine tuning


I am following the following script to fine tune the BERT base model on my dataset.

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:

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