simcse-pytorch

使用方法:

  • cd src/
  • 训练无监督simcse: python main.py --model_type="unsup"
  • 训练有监督simcse: python main.py --model_type="sup"

博客链接: 句向量模型之SimCSE——Pytorch