Forráskód Böngészése

add SOLOv2 to trt block list

gaotingquan 8 hónapja
szülő
commit
03d12bd2f6
1 módosított fájl, 1 hozzáadás és 0 törlés
  1. 1 0
      paddlex/inference/utils/trt_blacklist.py

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

@@ -37,4 +37,5 @@ TRT_BLOCKLIST = [
     "Cascade-MaskRCNN-ResNet50-FPN",
     "MaskRCNN-ResNet50-vd-FPN",
     "Cascade-MaskRCNN-ResNet50-FPN",
+    "SOLOv2",
 ]