yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-26 21:24:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

[TIPC]update tipc scripts (#11097)

Browse Source
This commit is contained in:
Kai Song 2023-10-18 20:05:52 +08:00 committed by GitHub
parent e3fc6393e0
commit a0218a8b83
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test_tipc/test_train_inference_python_npu.sh
View File

@ -43,6 +43,10 @@ if [ $modelname == "rec_r31_sar" ] || [ $modelname == "rec_mtb_nrtr" ]; then
sed -i "s/gpu_list:0|0,1/gpu_list:0,1/g" $FILENAME
sed -i "s/Global.use_npu:True|True/Global.use_npu:True/g" $FILENAME
fi
if [ $modelname == "ch_ppocr_mobile_v2_0_rec_FPGM" ]; then
sed -i '18s/$/ -o Global.use_gpu=False/' $FILENAME
sed -i '32s/$/ Global.use_gpu=False/' $FILENAME
fi
# replace training config file
grep -n 'tools/.*yml' $FILENAME | cut -d ":" -f 1 \

4
test_tipc/test_train_inference_python_xpu.sh
View File

@ -43,6 +43,10 @@ if [ $modelname == "rec_r31_sar" ] || [ $modelname == "rec_mtb_nrtr" ]; then
sed -i "s/gpu_list:0|0,1/gpu_list:0,1/g" $FILENAME
sed -i "s/Global.use_xpu:True|True/Global.use_xpu:True/g" $FILENAME
fi
if [ $modelname == "ch_ppocr_mobile_v2_0_rec_FPGM" ]; then
sed -i '18s/$/ -o Global.use_gpu=False/' $FILENAME
sed -i '32s/$/ Global.use_gpu=False/' $FILENAME
fi
# replace training config file
grep -n 'tools/.*yml' $FILENAME | cut -d ":" -f 1 \