diff --git a/tests/ocr_det_params.txt b/tests/ocr_det_params.txt index be29a0f5d9f8a895c7f0e58e2e7c79a71409e5ca..f14da857382827580f3659e726db6c64d7280ad1 100644 --- a/tests/ocr_det_params.txt +++ b/tests/ocr_det_params.txt @@ -1,7 +1,7 @@ ===========================train_params=========================== model_name:ocr_det python:python3.7 -gpu_list:3 +gpu_list:0|0,1 Global.use_gpu:True|True Global.auto_cast:null Global.epoch_num:lite_train_infer=1|whole_train_infer=300 diff --git a/tests/ocr_det_params_prune.txt b/tests/ocr_det_params_prune.txt deleted file mode 100644 index de218c1b39013ff1bde08ec592e791c97d801b01..0000000000000000000000000000000000000000 --- a/tests/ocr_det_params_prune.txt +++ /dev/null @@ -1,52 +0,0 @@ -===========================train_params=========================== -model_name:ocr_det -python:python3.7 -gpu_list:3 -Global.use_gpu:True -Global.auto_cast:null -Global.epoch_num:lite_train_infer=1|whole_train_infer=300 -Global.save_model_dir:./output/ -Train.loader.batch_size_per_card:lite_train_infer=2|whole_train_infer=4 -Global.pretrained_model:null -train_model_name:latest -train_infer_img_dir:./train_data/icdar2015/text_localization/ch4_test_images/ -null:null -## -trainer:fpgm_train -norm_train:null -pact_train:null -fpgm_train:deploy/slim/prune/sensitivity_anal.py -c configs/det/det_mv3_db.yml -o -distill_train:null -null:null -null:null -## -===========================eval_params=========================== -eval:tools/eval.py -c configs/det/det_mv3_db.yml -o -null:null -## -===========================infer_params=========================== -Global.save_inference_dir:./output/ -Global.pretrained_model: -norm_export:tools/export_model.py -c configs/det/det_mv3_db.yml -o -quant_export:deploy/slim/quantization/export_model.py -c configs/det/det_mv3_db.yml -o -fpgm_export:deploy/slim/prune/export_prune_model.py -c configs/det/det_mv3_db.yml -o -distill_export:null -export1:null -export2:null -## -infer_model:./inference/ch_ppocr_mobile_v2.0_det_infer/ -infer_export:null -infer_quant:False -inference:tools/infer/predict_det.py ---use_gpu:True|False ---enable_mkldnn:True|False ---cpu_threads:1|6 ---rec_batch_num:1 ---use_tensorrt:False|True ---precision:fp32|fp16|int8 ---det_model_dir: ---image_dir:./inference/ch_det_data_50/all-sum-510/ ---save_log_path:null ---benchmark:True -null:null -