Selaa lähdekoodia

update trt config for PP-FormulaNet-L

gaotingquan 8 kuukautta sitten
vanhempi
commit
a8cd928d1e
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      paddlex/inference/utils/trt_config.py

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

@@ -149,7 +149,7 @@ PIR_TRT_CFG_SETTING = {
     "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,
+        "workspace_size": 2 << 32,
     },
     "PP-FormulaNet-S": {
         "disable_ops": ["pd_op.full_with_tensor"],