Install first

pip3 install -e .

Training

cd examples/training/similarity_search
python3 training_similarity_model.py

Acknowledgements

Codes are adapted from the repos of the EMNLP19 paper Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks and the ICLR19 paper Learning Deep Representations by Mutual Information Estimation and Maximization