yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-01-03 10:37:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #7266 from WenmuZhou/tipc

Browse Source 
add PP-OCRv2 det amp custom_black_list
This commit is contained in:
zhoujun 2022-08-19 17:02:24 +08:00 committed by GitHub
commit 94710ae319
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml
View File

@ -14,6 +14,9 @@ Global:
use_visualdl: False
infer_img: doc/imgs_en/img_10.jpg
save_res_path: ./output/det_db/predicts_db.txt
use_amp: False
amp_level: O2
amp_custom_black_list: ['exp']
Architecture:
name: DistillationModel

3
tools/program.py
View File

@ -278,7 +278,8 @@ def train(config,
model_average = True
# use amp
if scaler:
with paddle.amp.auto_cast(level=amp_level):
custom_black_list = config['Global'].get('amp_custom_black_list',[])
with paddle.amp.auto_cast(level=amp_level, custom_black_list=custom_black_list):
if model_type == 'table' or extra_input:
preds = model(images, data=batch[1:])
elif model_type in ["kie", 'vqa']: