ckiplab / ckip-transformers

CKIP Transformers

Home Page:https://ckip-transformers.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

請問finetune模型前,有辦法更改模型原有label的個數嗎

andylee40 opened this issue · comments

您好,我想要微調ckiplab/bert-base-chinese-ner這個模型,但看到模型的label有72個,有辦法從72個label中選我會使用到的29個,然後再進行微調嗎?

你可以直接忽視沒在用的 label。
如果在乎效率或空間問題,可以修改最後一層的 linear layer 把用不到的 weight 刪掉。