diff --git a/mmdeploy/apis/tensorrt/tensorrt_utils.py b/mmdeploy/apis/tensorrt/tensorrt_utils.py index 2533e9541..d867de2f6 100644 --- a/mmdeploy/apis/tensorrt/tensorrt_utils.py +++ b/mmdeploy/apis/tensorrt/tensorrt_utils.py @@ -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 diff --git a/tools/deploy.py b/tools/deploy.py index 7a1095669..768098783 100644 --- a/tools/deploy.py +++ b/tools/deploy.py @@ -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):