From dd2f7de43174a15c43dd61543f0a1b9b2aaa0769 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Wed, 24 Aug 2022 05:53:42 +0000 Subject: [PATCH] fix name --- ppocr/modeling/backbones/vqa_layoutlm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppocr/modeling/backbones/vqa_layoutlm.py b/ppocr/modeling/backbones/vqa_layoutlm.py index ec64b032..d17a3883 100644 --- a/ppocr/modeling/backbones/vqa_layoutlm.py +++ b/ppocr/modeling/backbones/vqa_layoutlm.py @@ -29,14 +29,14 @@ __all__ = ["LayoutXLMForSer", "LayoutLMForSer"] pretrained_model_dict = { LayoutXLMModel: { "base": "layoutxlm-base-uncased", - "vi": "layoutxlm-wo-backbone-base-uncased", + "vi": "vi-layoutxlm-base-uncased", }, LayoutLMModel: { "base": "layoutlm-base-uncased", }, LayoutLMv2Model: { "base": "layoutlmv2-base-uncased", - "vi": "layoutlmv2-wo-backbone-base-uncased", + "vi": "vi-layoutlmv2-backbone-base-uncased", }, } -- GitLab