bert 分类模型
大神的帖子:https://blog.csdn.net/xmxoxo/article/details/89315370github:https://github.com/xmxoxo/BERT-train2deploy记录本人训练情况:1.数据:train.tsv,test.tsv ,dev.tsv2.改源码:-> 修改run_classifier.py文件中get_lab...
大神的帖子:https://blog.csdn.net/xmxoxo/article/details/89315370
github: https://github.com/xmxoxo/BERT-train2deploy
记录本人训练情况:
1.数据:train.tsv,test.tsv ,dev.tsv
2.改源码:-> 修改run_classifier.py文件中get_labels等方法
3.保存label2id的字典
4.使用命令训练模型
5.预测
6.后面使用的模型服务端,使用的是.pb格式的模型文件,所以需要把生成的ckpt格式模型文件转换成.pb格式的模型文件。
7. 部署服务器bert-base-serving-start -model_dir /icooper/output/mobile_0 -bert_model_dir /icooper/chinese_L-12_H-768_A-12 -model_pb_dir /icooper/output/mobile_0 -mode CLASS -max_seq_len 200 -port 5575 -port_out 5576 -http_port 8091
新手踩点:
1.当时在docker中部署环境LANG="" 编码不对,locale查看,改LANG=en_US.UTF-8
2.没保存pd文件bert-base-serving-start运行出来有莫名其妙的错误,保存后正确
果然跟大神走是没错的!!!
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
所有评论(0)