add db++

872d73da · wangjingyeye · d915416d · 872d73da
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

tools/program.py tools/program.py +2 -2

未找到文件。
--- a/tools/program.py
+++ b/tools/program.py
@@ -574,8 +574,8 @@ def preprocess(is_train=False):
    assert alg in [
        'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN',
        'CLS', 'PGNet', 'Distillation', 'NRTR', 'TableAttn', 'SAR', 'PSE',
-        'SEED', 'SDMGR', 'LayoutXLM', 'LayoutLM', 'PREN', 'FCE', 'SVTR',
-        'ViTSTR', 'ABINet', 'DB++'
+        'SEED', 'SDMGR', 'LayoutXLM', 'LayoutLM', 'LayoutLMv2', 'PREN', 'FCE',
+        'SVTR', 'ViTSTR', 'ABINet', 'DB++'
    ]

    if use_xpu: