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

3 lines
102 B
Python
Raw Normal View History

backend_config = dict(
2021-12-13 19:57:15 +08:00
type='tensorrt', common_config=dict(fp16_mode=True, max_workspace_size=0))