Merge pull request #40 from LDOUBLEV/fixocr

solve predicts diff between cpu and gpu running
2020-05-15 17:33:52 +08:00 · 2020-05-15 17:33:52 +08:00 · dc20feb13a
parent e807901b7b 7259370cbd
commit dc20feb13a
1 changed files with 1 additions and 0 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -91,6 +91,7 @@ def create_predictor(args, mode):
    config.disable_glog_info()

    # use zero copy
+    config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
    config.switch_use_feed_fetch_ops(False)
    predictor = create_paddle_predictor(config)
    input_names = predictor.get_input_names()