From 21da7f4c00aac63b2ea6d4ab875ecb60e62f52a4 Mon Sep 17 00:00:00 2001 From: Sijun He Date: Fri, 18 Aug 2023 11:51:54 +0800 Subject: [PATCH] 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 --- ppocr/modeling/backbones/vqa_layoutlm.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/ppocr/modeling/backbones/vqa_layoutlm.py b/ppocr/modeling/backbones/vqa_layoutlm.py index acb1315cc0..4357b56645 100644 --- a/ppocr/modeling/backbones/vqa_layoutlm.py +++ b/ppocr/modeling/backbones/vqa_layoutlm.py @@ -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