search

princeton-nlp / SimCSE

[EMNLP 2021] SimCSE: Simple Contrastive Learning of Sentence Embeddings https://arxiv.org/abs/2104.08821
MIT License
3.36k stars 507 forks source link

fine tune sup-SimCSE #173

Closed DanHuang752 closed 2 years ago

DanHuang752 commented 2 years ago

Hi,

Thanks for the great work! I would like to fine tune sup-SimCSE by my own dataset. Can I set 'model_name_or_path' in run_sup_example.sh to princeton-nlp/sup-simcse-bert-base-uncased, set train_file to my own training data to fine tune sup-SimCSE ? Thanks in advance!

gaotianyu1350 commented 2 years ago

Hi,

Yes you can do so!

DanHuang752 commented 2 years ago

Thanks for the quick reply!

github-actions[bot] commented 2 years ago

Stale issue message