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 # create engine
with torch.cuda.device(device): with torch.cuda.device(device):
engine = builder.build_engine(network, config) 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 return engine

3
tools/deploy.py
View File

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