曾寫過bert cs服務方式的調用方法:https://blog.csdn.net/renyuanfang/article/details/86701148 ,這種方式優點是操作簡單,但也有很大的缺陷,不能 fine-tune,操作繁瑣,而且速度非常慢 ,不能在windows上使用。
最近一直嘗試另一種bert的調用方式,克服了上面所說缺點,具體如下:
1.準備
tensorflow 版本爲 1.13,1.14,1.15 都可以
參考 https://blog.csdn.net/renyuanfang/article/details/86701148 ,進行預訓練模型下載
2.下載代碼
git 地址:https://github.com/HappyYuanFang/bert_base.git
其中模型文件未放到git中,具體目錄結構如下(確定目錄能對應上):
3.測試代碼
hello_bert.py 具體如下
windows 測試結果