solve predicts diff between cpu and gpu running

2020-05-15 17:32:32 +08:00 · 2020-05-15 17:32:32 +08:00 · 7259370cbd
parent 1e9683b426
commit 7259370cbd
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()