提交 7259370c 编写于 作者: L LDOUBLEV

solve predicts diff between cpu and gpu running

上级 1e9683b4
......@@ -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()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册