Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
大约 1 年 前同步成功

通知 115
Star 4999
Fork 1114
P
PaddleClas
提交 6a55aac3 编写于 作者: D duanyanhui 提交者： cuicheng01
浏览文件
操作

update npu inference api

上级 a89269e5
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
deploy/utils/predictor.py
浏览文件 @ 6a55aac3
...@@ -63,7 +63,7 @@ class Predictor(object): ...@@ -63,7 +63,7 @@ class Predictor(object):
if args.get("use_gpu", False): if args.get("use_gpu", False):
config.enable_use_gpu(args.gpu_mem, 0) config.enable_use_gpu(args.gpu_mem, 0)
elif args.get("use_npu", False): elif args.get("use_npu", False):
config.enable_npu() config.enable_custom_device('npu')
elif args.get("use_xpu", False): elif args.get("use_xpu", False):
config.enable_xpu() config.enable_xpu()
else: else:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或