Sfoglia il codice sorgente

Fix reshape and layer_norm layer with ch_SVTRv2_rec

zhink 8 mesi fa
parent
commit
45313d80a6
1 ha cambiato i file con 21 aggiunte e 0 eliminazioni
  1. 21 0
      paddlex/inference/utils/trt_config.py

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

@@ -166,6 +166,27 @@ OLD_IR_TRT_CFG_SETTING = {
             ]
         ],
     },
+    "ch_SVTRv2_rec": {
+        "enable_tensorrt_engine": OLD_IR_TRT_CFG_DEFAULT_SETTING,
+        "exp_disable_tensorrt_ops": [
+            [
+                "reshape2_3.tmp_0",
+                "reshape2_5.tmp_0",
+                "reshape2_7.tmp_0",
+                "reshape2_9.tmp_0",
+                "reshape2_11.tmp_0",
+                "reshape2_13.tmp_0",
+                "reshape2_15.tmp_0",
+                "reshape2_17.tmp_0",
+                "reshape2_19.tmp_0",
+                "reshape2_28.tmp_0",
+                "reshape2_42.tmp_0",
+                "reshape2_47.tmp_0",
+                "layer_norm_15.tmp_2",
+                "layer_norm_13.tmp_2",
+            ]
+        ],
+    },
 }
 
 ############ pir trt ############