mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

replace gpu id

pull/3945/head
tink2123 2021-08-19 08:58:35 +00:00
parent 418385b88c
commit ec300c652d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/ocr_rec_params.txt
View File

@ -1,7 +1,7 @@
===========================train_params=========================== ===========================train_params===========================
model_name:ocr_rec model_name:ocr_rec
python:python3.7 python:python3.7
gpu_list:0|2,3 gpu_list:0|0,1
Global.use_gpu:True|True Global.use_gpu:True|True
Global.auto_cast:null Global.auto_cast:null
Global.epoch_num:lite_train_infer=2|whole_train_infer=300 Global.epoch_num:lite_train_infer=2|whole_train_infer=300
@ -41,7 +41,7 @@ inference:tools/infer/predict_rec.py
--use_gpu:True|False --use_gpu:True|False
--enable_mkldnn:True|False --enable_mkldnn:True|False
--cpu_threads:1|6 --cpu_threads:1|6
--rec_batch_num:1 --rec_batch_num:1|6
--use_tensorrt:True|False --use_tensorrt:True|False
--precision:fp32|fp16|int8 --precision:fp32|fp16|int8
--rec_model_dir: --rec_model_dir: