6 lines
138 B
Python
Raw Normal View History

2021-06-23 13:14:28 +08:00
import tensorrt as trt
2021-06-17 15:28:23 +08:00
backend = 'tensorrt'
2021-06-23 13:14:28 +08:00
tensorrt_param = dict(
shared_param=dict(log_level=trt.Logger.WARNING, fp16_mode=False))