Browse Source

disable slice op in trt for SLANeXt_wireless presicion

gaotingquan 8 tháng trước cách đây
mục cha
commit
eddbbe061f
1 tập tin đã thay đổi với 1 bổ sung0 xóa
  1. 1 0
      paddlex/inference/utils/trt_config.py

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

@@ -136,6 +136,7 @@ PIR_TRT_PRECISION_MAP = PIR_TRT_PRECISION_MAP_CLASS()
 
 PIR_TRT_CFG_SETTING = {
     "SLANeXt_wired": {"disable_ops": ["pd_op.slice"]},
+    "SLANeXt_wireless": {"disable_ops": ["pd_op.slice"]},
     "DETR-R50": {"optimization_level": 4, "workspace_size": 1 << 32},
     "SegFormer-B0": {"optimization_level": 4, "workspace_size": 1 << 32},
     "SegFormer-B1": {"optimization_level": 4, "workspace_size": 1 << 32},