Skip to content

Commit 39d0402

Browse files
committed
nlp bug
1 parent 09fafed commit 39d0402

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: example/auto_compression/nlp/paddle_inference_eval.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ def create_predictor(cls, args):
238238
config.enable_tensorrt_engine(
239239
workspace_size=1 << 30,
240240
max_batch_size=args.batch_size,
241-
min_subgraph_size=76,
241+
min_subgraph_size=5,
242242
precision_mode=precision_map[args.precision],
243243
use_static=True,
244244
use_calib_mode=False, )

0 commit comments

Comments
 (0)