未验证 提交 21da7f4c 编写于 作者: S Sijun He 提交者: GitHub

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
上级 dbf35bb7
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册