From 7259370cbd5db7e3ba15ec821a279ad1d106045c Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Fri, 15 May 2020 17:32:32 +0800 Subject: [PATCH] solve predicts diff between cpu and gpu running --- tools/infer/utility.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index ff458787..cb7833a1 100755 --- 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() -- GitLab