mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1492 from cuicheng01/release/2.3

Browse Source 
[cherry-pick]update predictor.py
This commit is contained in:
cuicheng01 2021-11-30 11:10:59 +08:00 committed by GitHub
commit da3aefada0
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
deploy/utils/predictor.py
View File

@ -60,6 +60,7 @@ class Predictor(object):
precision_mode=Config.Precision.Half
if args.use_fp16 else Config.Precision.Float32,
max_batch_size=args.batch_size,
workspace_size=1 << 30,
min_subgraph_size=30)
config.enable_memory_optim()