zjunlp / KnowPrompt

[WWW 2022] KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation Extraction
MIT License
194 stars 34 forks source link

关于预训练语言模型选择的问题 #10

Closed Real-bojack closed 2 years ago

Real-bojack commented 2 years ago

请问一下作者,如果我想选择bert作为预训练模型,应该修改哪些参数?

njcx-ai commented 2 years ago

很抱歉当初实验时候只跑了roberta, 代码内没设置模型切换,若想选择bert作为预训练模型可能得改一下代码细节而不仅是输入参数。

Real-bojack commented 2 years ago

哦哦 谢谢!