Update utility.py

5 years ago · 51b57c47c7
parent 04c5c6ab3b
commit 51b57c47c7
1 changed files with 3 additions and 0 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -84,6 +84,9 @@ def parse_args():

    parser.add_argument("--enable_mkldnn", type=str2bool, default=False)
    parser.add_argument("--use_zero_copy_run", type=str2bool, default=False)
+    
+    parser.add_argument("--use_pdserving", type=str2bool, default=False)
+    
    return parser.parse_args()