Skip to content

GitLab

"ppstructure/vscode:/vscode.git/clone" did not exist on "2f9a47ca6ab48c3e7356ca1e5a0c02c8276a5a7f"
Commit 82f19a31 authored by stephon's avatar stephon
Browse files

support cpu train in script

parent 987517fd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/configs/ppocr_det_mobile_params.txt
View file @ 82f19a31
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
model_name:ocr_det model_name:ocr_det
python:python3.7 python:python3.7
gpu_list:0|0,1|10.21.226.181,10.21.226.133;0,1 gpu_list:0|0,1|10.21.226.181,10.21.226.133;0,1
Global.use_gpu:True|True Global.use_gpu:False|True
Global.auto_cast:fp32|amp Global.auto_cast:fp32|amp
Global.epoch_num:lite_train_infer=1|whole_train_infer=300 Global.epoch_num:lite_train_infer=1|whole_train_infer=300
Global.save_model_dir:./output/ Global.save_model_dir:./output/
......
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