Commit 45f50826 authored by myhloli's avatar myhloli
Browse files

refactor(ocr): update device parameter handling in paddleocr2pytorch

- Replace get_device() function call with direct 'device' variable usage
- Simplify device configuration in OCR model initialization
parent 4f88fcaa
......@@ -80,7 +80,7 @@ class PytorchPaddleOCR(TextSystem):
kwargs['rec_char_dict_path'] = os.path.join(root_dir, 'pytorchocr', 'utils', 'resources', 'dict', dict_file)
# kwargs['rec_batch_num'] = 8
kwargs['device'] = get_device()
kwargs['device'] = device
default_args = vars(args)
default_args.update(kwargs)
......
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