diff --git a/tests/ocr_rec_params.txt b/tests/ocr_rec_params.txt index 123719f95110d1aeacef3ff8810438802ec4e4a3..2d27a9ee3b97364cf0eb95ce44f86456f493b995 100644 --- a/tests/ocr_rec_params.txt +++ b/tests/ocr_rec_params.txt @@ -1,7 +1,7 @@ ===========================train_params=========================== model_name:ocr_rec python:python3.7 -gpu_list:0|2,3 +gpu_list:0|0,1 Global.use_gpu:True|True Global.auto_cast:null 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 --enable_mkldnn:True|False --cpu_threads:1|6 ---rec_batch_num:1 +--rec_batch_num:1|6 --use_tensorrt:True|False --precision:fp32|fp16|int8 --rec_model_dir: