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

Merge pull request #557 from Sunny-wong/develop 

Update dataset_traversal.py
This commit is contained in:
MissPenguin 2020-08-19 13:52:23 +08:00 committed by GitHub
parent d2e1eb5e9c b2e1a84981
commit 8ad1cbe01d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppocr/data/rec/dataset_traversal.py
View File

@ -237,7 +237,7 @@ class SimpleReader(object):
def get_device_num(): def get_device_num():
if self.use_gpu: if self.use_gpu:
gpus = os.environ.get("CUDA_VISIBLE_DEVICES", 1) gpus = os.environ.get("CUDA_VISIBLE_DEVICES", '1')
gpu_num = len(gpus.split(',')) gpu_num = len(gpus.split(','))
return gpu_num return gpu_num
else: else: