search

fastnlp / fastNLP

fastNLP: A Modularized and Extensible NLP Framework. Currently still in incubation.
https://gitee.com/fastnlp/fastNLP
Apache License 2.0
3.05k stars 451 forks source link

文本相似度模型（BertForSentenceMatching）可以使用STSbenchmark数据集进行训练么 #320

Closed jwc19890114 closed 3 years ago

jwc19890114 commented 3 years ago

你好，想请教一下文本相似度模型（BertForSentenceMatching）可以使用STSbenchmark数据集进行训练么。如果可以的话是不是需要将forward方法中的argmax部分进行调整呢

xuyige commented 3 years ago

可以的 可能不需要调整forward部分，只需要调整predict部分就可以了