From 5b8750b83bb6871bba2d27c0f33cb66a462c09da Mon Sep 17 00:00:00 2001 From: RunningLeon Date: Fri, 24 Sep 2021 19:09:39 +0800 Subject: [PATCH] [Enhance]: Logging trace info for debugging in deploy tool (#93) * print trace info for debugging * resolve comment --- mmdeploy/apis/tensorrt/tensorrt_utils.py | 3 ++- tools/deploy.py | 3 +++ 2 files changed, 5 insertions(+), 1 deletion(-) 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):