Skip to content

Commit eb1dd6d

Browse files
authored
fix min_sub_graph (#1606)
1 parent df48dea commit eb1dd6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_tipc/bigru_crf/deploy/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ def __init__(self,
173173
if use_tensorrt:
174174
config.enable_tensorrt_engine(
175175
max_batch_size=batch_size,
176-
min_subgraph_size=0,
176+
min_subgraph_size=1,
177177
precision_mode=precision_mode,
178178
use_calib_mode=use_calib_mode)
179179
min_input_shape = {

0 commit comments

Comments
 (0)