mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
7 lines
166 B
Python
7 lines
166 B
Python
import tensorrt as trt
|
|
|
|
backend_config = dict(
|
|
type='tensorrt',
|
|
common_config=dict(
|
|
fp16_mode=True, log_level=trt.Logger.INFO, max_workspace_size=0))
|