Skip to content

GitLab

"ml/backend/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "718961de68fb82f355bca53c1bdc3d126a42fb86"
Commit ec300c65 authored by tink2123's avatar tink2123
Browse files

replace gpu id

parent 418385b8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/ocr_rec_params.txt
View file @ ec300c65
===========================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 ...@@ -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:
......
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