support cpu train in script

82f19a31 · stephon · 987517fd · 82f19a31
Commit 82f19a31 authored Oct 15, 2021 by stephon
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/configs/ppocr_det_mobile_params.txt tests/configs/ppocr_det_mobile_params.txt +1 -1

No files found.
--- a/tests/configs/ppocr_det_mobile_params.txt
+++ b/tests/configs/ppocr_det_mobile_params.txt
@@ -2,7 +2,7 @@
 model_name:ocr_det
 python:python3.7
 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.epoch_num:lite_train_infer=1|whole_train_infer=300
 Global.save_model_dir:./output/