From 9682025ce99b09115163af6d15c928911acf6692 Mon Sep 17 00:00:00 2001 From: sjtubinlong Date: Sat, 14 Dec 2019 21:13:59 +0800 Subject: [PATCH] add use_pr config item in export_model --- pdseg/export_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pdseg/export_model.py b/pdseg/export_model.py index 410cfc33..93e4b494 100644 --- a/pdseg/export_model.py +++ b/pdseg/export_model.py @@ -52,6 +52,7 @@ def parse_args(): def export_inference_config(): deploy_cfg = '''DEPLOY: USE_GPU : 1 + USE_PR : 1 MODEL_PATH : "%s" MODEL_FILENAME : "%s" PARAMS_FILENAME : "%s" -- GitLab