瀏覽代碼

set PP-YOLOE_plus_SOD-largesize-L trt config

gaotingquan 8 月之前
父節點
當前提交
d3edb80885
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      paddlex/inference/utils/trt_config.py

+ 1 - 0
paddlex/inference/utils/trt_config.py

@@ -135,6 +135,7 @@ OLD_IR_TRT_CFG_SETTING = {
 PIR_TRT_PRECISION_MAP = PIR_TRT_PRECISION_MAP_CLASS()
 
 PIR_TRT_CFG_SETTING = {
+    "PP-YOLOE_plus_SOD-largesize-L": {"workspace_size": 1 << 32},
     "SLANeXt_wired": {"disable_ops": ["pd_op.slice"]},
     "SLANeXt_wireless": {"disable_ops": ["pd_op.slice"]},
     "DETR-R50": {"optimization_level": 4, "workspace_size": 1 << 32},