浏览代码

set trt setting for PP-FormulaNet

gaotingquan 8 月之前
父节点
当前提交
af8c83f73b
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      paddlex/inference/utils/trt_config.py

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

@@ -22,4 +22,12 @@ TRT_CFG = {
     "SegFormer-B5": {"optimization_level": 4, "workspace_size": 1 << 32},
     "LaTeX_OCR_rec": {"disable_ops": ["pd_op.slice"]},
     "PP-YOLOE_seg-S": {"disable_ops": ["pd_op.slice", "pd_op.bilinear_interp"]},
+    "PP-FormulaNet-L": {
+        "disable_ops": ["pd_op.full_with_tensor"],
+        "workspace_size": 1 << 32,
+    },
+    "PP-FormulaNet-S": {
+        "disable_ops": ["pd_op.full_with_tensor"],
+        "workspace_size": 1 << 32,
+    },
 }