Skip to content

GitLab

Unverified Commit e0d7deaa authored by Thomas Young's avatar Thomas Young Committed by GitHub
Browse files

fix gpu_id 

fix gpu_id
parent 6893d151
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
deploy/pdserving/win/ocr_web_server.py
View file @ e0d7deaa
...@@ -42,7 +42,7 @@ class OCRService(WebService): ...@@ -42,7 +42,7 @@ class OCRService(WebService):
self.det_client = LocalPredictor() self.det_client = LocalPredictor()
if sys.argv[1] == 'gpu': if sys.argv[1] == 'gpu':
self.det_client.load_model_config( self.det_client.load_model_config(
det_model_config, use_gpu=True, gpu_id=1) det_model_config, use_gpu=True, gpu_id=0)
elif sys.argv[1] == 'cpu': elif sys.argv[1] == 'cpu':
self.det_client.load_model_config(det_model_config) self.det_client.load_model_config(det_model_config)
self.ocr_reader = OCRReader( self.ocr_reader = OCRReader(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment