mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Supplemental a parameter "device" (#1165) 

* Supplemental a parameter "device"
补充一个参数“device”

* 补充一个参数“device”
Supplemental a parameter "device"
pull/1204/head
达芬奇 2022-10-09 14:26:14 +08:00 committed by GitHub
parent b2cb5722f7
commit 895d0749ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/en/02-how-to-run/prebuilt_package_windows.md
View File

@ -163,7 +163,7 @@ torch2onnx(img, work_dir, save_file, deploy_cfg, model_cfg,
model_checkpoint, device)
# 2. extract pipeline info for sdk use (dump-info)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint, device=device)
```
The structure of the converted model directory:
@ -220,7 +220,7 @@ device = 'cuda'
onnx2tensorrt(work_dir, save_file, model_id, deploy_cfg, onnx_model, device)
# 3. extract pipeline info for sdk use (dump-info)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint, device=device)
```
The structure of the converted model directory:

4
docs/zh_cn/02-how-to-run/prebuilt_package_windows.md
View File

@ -171,7 +171,7 @@ torch2onnx(img, work_dir, save_file, deploy_cfg, model_cfg,
model_checkpoint, device)
# 2. extract pipeline info for sdk use (dump-info)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint, device=device)
```
转换后的模型目录结构应该为：
@ -228,7 +228,7 @@ device = 'cuda'
onnx2tensorrt(work_dir, save_file, model_id, deploy_cfg, onnx_model, device)
# 3. extract pipeline info for sdk use (dump-info)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint)
export2SDK(deploy_cfg, model_cfg, work_dir, pth=model_checkpoint, device=device)
```
转换后的模型目录结构应该为：