mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update VQA to use the updated LayoutLM syntax from PaddleNLP (#9791) 

* Update vqa_layoutlm.py 

Update vqa to use the updated LayoutLM syntax from PaddleNLP

* Update vqa_layoutlm.py
pull/10654/head
Sijun He 2023-08-18 11:51:54 +08:00 committed by GitHub
parent dbf35bb714
commit 21da7f4c00
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ppocr/modeling/backbones/vqa_layoutlm.py
View File

@ -54,18 +54,11 @@ class NLPBaseModel(nn.Layer):
if checkpoints is not None: # load the trained model
self.model = model_class.from_pretrained(checkpoints)
else: # load the pretrained-model
pretrained_model_name = pretrained_model_dict[base_model_class][
mode]
if pretrained is True:
base_model = base_model_class.from_pretrained(
pretrained_model_name)
else:
base_model = base_model_class.from_pretrained(pretrained)
pretrained_model_name = pretrained_model_dict[base_model_class][mode]
if type == "ser":
self.model = model_class(
base_model, num_classes=kwargs["num_classes"], dropout=None)
self.model = model_class.from_pretrained(pretrained_model_name, num_classes=kwargs["num_classes"], dropout=0)
else:
self.model = model_class(base_model, dropout=None)
self.model = model_class.from_pretrained(pretrained_model_name, dropout=0)
self.out_channels = 1
self.use_visual_backbone = True