mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

[Enhance]: Logging trace info for debugging in deploy tool (#93)

Browse Source 
* print trace info for debugging

* resolve comment
This commit is contained in:
RunningLeon 2021-09-24 19:09:39 +08:00 committed by GitHub
parent de096d5f00
commit 5b8750b83b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmdeploy/apis/tensorrt/tensorrt_utils.py
View File

@ -107,7 +107,8 @@ def create_trt_engine(onnx_model: Union[str, onnx.ModelProto],
# create engine
with torch.cuda.device(device):
engine = builder.build_engine(network, config)
assert engine is not None, f'Failed to create engine from {onnx_model}'
assert engine is not None, 'Failed to create TensorRT engine'
return engine

3
tools/deploy.py
View File

@ -2,6 +2,8 @@ import argparse
import logging
import os.path as osp
import subprocess
import sys
import traceback
from functools import partial
import mmcv
@ -59,6 +61,7 @@ def target_wrapper(target, log_level, ret_value, *args, **kwargs):
return result
except Exception as e:
logging.error(e)
traceback.print_exc(file=sys.stdout)
def create_process(name, target, args, kwargs, ret_value=None):