fix min_sub_graph (#1606)

PaddlePaddle · Jan 20, 2022 · eb1dd6d · eb1dd6d
1 parent df48dea
commit eb1dd6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_tipc/bigru_crf/deploy/predict.py b/tests/test_tipc/bigru_crf/deploy/predict.py
@@ -173,7 +173,7 @@ def __init__(self,
             if use_tensorrt:
                 config.enable_tensorrt_engine(
                     max_batch_size=batch_size,
-                    min_subgraph_size=0,
+                    min_subgraph_size=1,
                     precision_mode=precision_mode,
                     use_calib_mode=use_calib_mode)
                 min_input_shape = {