replace gpu id

ec300c65 · tink2123 · 418385b8 · ec300c65
Commit ec300c65 authored Aug 19, 2021 by tink2123
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/ocr_rec_params.txt tests/ocr_rec_params.txt +2 -2

No files found.
--- a/tests/ocr_rec_params.txt
+++ b/tests/ocr_rec_params.txt
 ===========================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: