configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml

@@ -18,6 +18,7 @@ Global:
   save_res_path: ./checkpoints/det_db/predicts_db.txt
   distributed: true
   d2s_train_image_shape: [3, -1, -1]
+  amp_dtype: bfloat16
 
 Architecture:
   name: DistillationModel
@@ -222,4 +223,4 @@ Eval:
     shuffle: False
     drop_last: False
     batch_size_per_card: 1 # must be 1
-    num_workers: 2
\ No newline at end of file
+    num_workers: 2

test_tipc/benchmark_train.sh

@@ -96,7 +96,7 @@ model_name=$(func_parser_value "${lines[1]}")
 python_name=$(func_parser_value "${lines[2]}")
 
 # set env
-python=${python_name}
+python=python
 export str_tmp=$(echo `pip list|grep paddlepaddle-gpu|awk -F ' ' '{print $2}'`)
 export frame_version=${str_tmp%%.post*}
 export frame_commit=$(echo `${python} -c "import paddle;print(paddle.version.commit)"`)

test_tipc/prepare.sh

@@ -23,7 +23,7 @@ trainer_list=$(func_parser_value "${lines[14]}")
 if [ ${MODE} = "benchmark_train" ];then
     python_name_list=$(func_parser_value "${lines[2]}")
     array=(${python_name_list}) 
-    python_name=${array[0]}
+    python_name=python
     ${python_name} -m pip install -r requirements.txt
     if [[ ${model_name} =~ "ch_ppocr_mobile_v2_0_det" || ${model_name} =~ "det_mv3_db_v2_0" ]];then
         wget -nc -P  ./pretrain_models/ https://paddleocr.bj.bcebos.com/pretrained/MobileNetV3_large_x0_5_pretrained.pdparams  --no-check-certificate