中文詞向量bert預訓練模型下載地址
bert-serving-start文件的位置:home/anaconda3/envs/tensorflow/bin
cd home/anaconda3/envs/tensorflow/bin
bert-serving-start -model_dir /home/models/chinese_L-12_H-768_A-12 -num_worker=2
報錯:self.graph_path, self.bert_config = pool.apply(optimize_graph, (self.args,))
TypeError: cannot unpack non-iterable NoneType object
1、查看自己啓動代碼是否正確,-model_dir 後面跟自己下載的語料模型的解壓地址,注意是要進入文件夾裏面
2、檢查自己的TensorFlow的版本,2.0以上好像暫時還不支持
然後,還是啓動不了,查了好多帖子都沒用
隔了一兩個小時又試了一次,就好了
啓動提示信息
I:[33mWORKER-0[0m:use device gpu: 0, load graph from C:\Users\Bruce\AppData\Local\Temp\tmp4ig9cvgj
I:[33mWORKER-1[0m:use device gpu: 0, load graph from C:\Users\Bruce\AppData\Local\Temp\tmp4ig9cvgj
I:[33mWORKER-1[0m:ready and listening!
I:[33mWORKER-0[0m:ready and listening!
I:[35mVENTILATOR[0m:all set, ready to serve request!
接着可以在python環境下寫入測試代碼:
from bert_serving.client import BertClient
bc = BertClient()
print(bc.encode(['中國', '美國']))
可得到詞向量爲(詞向量的維度爲768):