Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
tensorrt.py 103 Bytes
Newer Older
limm's avatar
add config module  
limm committed
1
2 
backend_config = dict(
    type='tensorrt', common_config=dict(fp16_mode=False, max_workspace_size=0))