replace gpu id

2021-08-19 08:58:35 +00:00 · 2021-08-19 08:58:35 +00:00 · ec300c652d
parent 418385b88c
commit ec300c652d
1 changed files with 2 additions and 2 deletions
--- 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: