Skip to content

Commit

Permalink
config.switch_ir_debug(True)关掉
Browse files Browse the repository at this point in the history
  • Loading branch information
lizexu123 committed Feb 1, 2024
1 parent 39d0402 commit 09905eb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion example/auto_compression/nlp/paddle_inference_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -210,7 +210,7 @@ def create_predictor(cls, args):
config = paddle.inference.Config(
os.path.join(args.model_path, args.model_filename),
os.path.join(args.model_path, args.params_filename))
config.switch_ir_debug(True)
# config.switch_ir_debug(True)
# 适用于ERNIE 3.0-Medium模型
# config.exp_disable_tensorrt_ops(["elementwise_add"])
# config.exp_disable_tensorrt_ops(["fused_embedding_eltwise_layernorm"])
Expand Down

0 comments on commit 09905eb

Please sign in to comment.