7 lines
166 B
Python
7 lines
166 B
Python
|
import tensorrt as trt
|
||
|
|
|
||
|
|
backend_config = dict(
|
||
|
|
type='tensorrt',
|
||
|
|
common_config=dict(
|
||
|
|
fp16_mode=True, log_level=trt.Logger.INFO, max_workspace_size=0))
|