Merge pull request #557 from Sunny-wong/develop

Update dataset_traversal.py
5 years ago · 8ad1cbe01d
parent d2e1eb5e9c b2e1a84981
commit 8ad1cbe01d
1 changed files with 1 additions and 1 deletions
--- a/ppocr/data/rec/dataset_traversal.py
+++ b/ppocr/data/rec/dataset_traversal.py
@ -237,7 +237,7 @@ class SimpleReader(object):

        def get_device_num():
            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(','))
                return gpu_num
            else: