Просмотр исходного кода

set trt config for LaTeX_OCR_rec

gaotingquan 9 месяцев назад
Родитель
Сommit
a83f6eda75
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      paddlex/inference/utils/trt_config.py

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

@@ -20,4 +20,5 @@ TRT_CFG = {
     "SegFormer-B3": {"optimization_level": 4, "workspace_size": 1 << 32},
     "SegFormer-B4": {"optimization_level": 4, "workspace_size": 1 << 32},
     "SegFormer-B5": {"optimization_level": 4, "workspace_size": 1 << 32},
+    "LaTeX_OCR_rec": {"disable_ops": ["pd_op.slice"]},
 }