forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #40 from LDOUBLEV/fixocr 

solve predicts diff between cpu and gpu running
This commit is contained in:
Double_V 2020-05-15 17:33:52 +08:00 committed by GitHub
parent e807901b7b 7259370cbd
commit dc20feb13a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tools/infer/utility.py
View File

@ -91,6 +91,7 @@ def create_predictor(args, mode):
config.disable_glog_info() config.disable_glog_info()
# use zero copy # use zero copy
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
config.switch_use_feed_fetch_ops(False) config.switch_use_feed_fetch_ops(False)
predictor = create_paddle_predictor(config) predictor = create_paddle_predictor(config)
input_names = predictor.get_input_names() input_names = predictor.get_input_names()