提交 e8fe6ebf 编写于 作者: M MRXLT

fix bert demo

上级 1b95bcec
......@@ -42,7 +42,7 @@ pip install paddle_serving_app
```
执行
```
cat data-c.txt | python bert_client.py
head data-c.txt | python bert_client.py --model bert_seq20_client/serving_client_conf.prototxt
```
启动client读取data-c.txt中的数据进行预测,预测结果为文本的向量表示(由于数据较多,脚本中没有将输出进行打印),server端的地址在脚本中修改。
......
......@@ -29,13 +29,13 @@ from paddle_serving_app import ChineseBertReader
args = benchmark_args()
reader = ChineseBertReader(max_seq_len=20)
reader = ChineseBertReader({"max_seq_len": 20})
fetch = ["pooled_output"]
endpoint_list = ["127.0.0.1:9292"]
client = Client()
client.load_client_config(args.model)
client.connect(endpoint_list)
for line in fin:
for line in sys.stdin:
feed_dict = reader.process(line)
result = client.predict(feed=feed_dict, fetch=fetch)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册