基于huggingface/transforms(pytorch)框架实现Bert文本分类
基于huggingface/transforms-PyTorch框架实现Bert文本分类背景项目结构安装依赖包数据与预训练模型数据预训练模型代码部分
背景
作者在使用bert_keras实现bert文本分类模型后发现训练时并不能使用GPU加速训练,因此想使用huggingface/transforms框架实现bert文本分类模型,但是由于不清楚其模型输入格式、API没有中文介绍等原因,在实现过程中还是废了不少力,因此想在此分享一下现成的代码供大家使用。https://github.com/huggingfa
原创
2020-08-26 15:40:34 ·
6529 阅读 ·
2 评论