search

princeton-nlp / SimCSE

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

Update train.py #210

Closed peregilk closed 1 year ago

peregilk commented 1 year ago

torch_xla.core.xla_model should be imported to be able to train on tpu

gaotianyu1350 commented 1 year ago

Thanks!