Browse Source

add CenterNet models to trt blocklist

gaotingquan 8 months ago
parent
commit
9633ff94c6
1 changed files with 2 additions and 0 deletions
  1. 2 0
      paddlex/inference/utils/trt_blacklist.py

+ 2 - 0
paddlex/inference/utils/trt_blacklist.py

@@ -38,4 +38,6 @@ TRT_BLOCKLIST = [
     "MaskRCNN-ResNet50-vd-FPN",
     "MaskRCNN-ResNet50-vd-FPN",
     "Cascade-MaskRCNN-ResNet50-FPN",
     "Cascade-MaskRCNN-ResNet50-FPN",
     "SOLOv2",
     "SOLOv2",
+    "CenterNet-DLA-34",
+    "CenterNet-ResNet50",
 ]
 ]