mmdeploy/configs/_base_/backends/tensorrt.py

3 lines
103 B
Python

backend_config = dict(
type='tensorrt', common_config=dict(fp16_mode=False, max_workspace_size=0))