提交 86fbb6aa 编写于 作者: 走神的阿圆's avatar 走神的阿圆

add bs stat

上级 ce198ca2
......@@ -172,6 +172,10 @@ 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.
先完成此消息的编辑!
想要评论请 注册