From 82f19a312fbc93084b776b1391f44f504031a608 Mon Sep 17 00:00:00 2001 From: stephon Date: Fri, 15 Oct 2021 09:39:13 +0000 Subject: [PATCH] support cpu train in script --- tests/configs/ppocr_det_mobile_params.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/configs/ppocr_det_mobile_params.txt b/tests/configs/ppocr_det_mobile_params.txt index be14865c..c65ad718 100644 --- 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/ -- GitLab