mmdeploy/configs/_base_/backends/tensorrt-fp16.py

3 lines
102 B
Python

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