未验证 提交 af0ac59e 编写于 作者: 走神的阿圆's avatar 走神的阿圆 提交者: GitHub

Add Bert Service statistic for hub serving (#362)

* add bs stat

* fix ci
上级 ce198ca2
......@@ -172,6 +172,9 @@ class ServingCommand(BaseCommand):
from paddle_gpu_serving.run import BertServer
bs = BertServer(with_gpu=args.use_gpu)
bs.with_model(model_name=args.modules[0])
CacheUpdater(
"hub_bert_service", module=args.modules[0],
version="0.0.0").start()
bs.run(gpu_index=args.gpu, port=int(args.port))
@staticmethod
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册