Skip to content

GitLab

"doc/vscode:/vscode.git/clone" did not exist on "fe775780e4bcd87bf9a97fcf5fb646193ee70c12"
Commit 6e1cfb05 authored by MissPenguin's avatar MissPenguin
Browse files

fix cpu infer error for table structure

parent 3f92af45
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tools/infer/utility.py
View file @ 6e1cfb05
...@@ -331,10 +331,11 @@ def create_predictor(args, mode, logger): ...@@ -331,10 +331,11 @@ def create_predictor(args, mode, logger):
config.disable_glog_info() config.disable_glog_info()
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass") config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
if mode == 'structure':
config.delete_pass("fc_fuse_pass") # not supported for table
config.switch_use_feed_fetch_ops(False) config.switch_use_feed_fetch_ops(False)
config.switch_ir_optim(True) config.switch_ir_optim(True)
if mode == 'structure':
config.switch_ir_optim(False)
# create predictor # create predictor
predictor = inference.create_predictor(config) predictor = inference.create_predictor(config)
input_names = predictor.get_input_names() input_names = predictor.get_input_names()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment