search

Vincent131499 / TextClassifier_Transformer

个人基于谷歌开源的BERT编写的文本分类器（基于微调方式），可自由加载NLP领域知名的预训练语言模型BERT、Bert-wwm、Roberta、ALBert以及ERNIE1.0
161 stars 46 forks source link

模型兼容 #6

Closed frostjsy closed 3 years ago

frostjsy commented 4 years ago

这个脚本可以直接加载项目中罗列的模型么？还是只契合roberta

frostjsy commented 4 years ago

我试了下，这个可以兼容bert,roberta,bert-wwm，Albert报错了

Vincent131499 commented 4 years ago

这个项目支持BERT系列所有模型。若要使用ALbert，请仔细阅读下readme文件说明： 注意：在使用AlBert时，请将该项目下的modeling.py文件更新为ALBert项目中下的modeling.py，而后在运行

frostjsy commented 4 years ago

好的，谢谢啊