mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
27880afdcd
* better retinanet support * prepare split export tensorrt * optimizer cfg * free anchor when static shape * fix docstring
6 lines
138 B
Python
6 lines
138 B
Python
import tensorrt as trt
|
|
|
|
backend = 'tensorrt'
|
|
tensorrt_param = dict(
|
|
shared_param=dict(log_level=trt.Logger.WARNING, fp16_mode=False))
|