Using this with roberta-base

yao8839836 / kg-bert

KG-BERT: BERT for Knowledge Graph Completion

Apache License 2.0

679 stars 141 forks source link

Using this with roberta-base #28

Open Saatviga opened 2 years ago

Saatviga commented 2 years ago

Hi,

If I want to use this with roberta-base pre-trained model, what are the parts that needs modification? creating features changes with instead of [CLS] [SEP] tags and later Roberta for sequence classification should be used I guess. Please let me know if there are further modifications required.