change support list (#14293)

pull/14297/head
liuhongen1234567 2024-11-29 15:18:54 +08:00 committed by GitHub
parent b6bcde143d
commit 8fdc409edf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions

View File

@ -41,7 +41,16 @@ def apply_to_static(model, config, logger):
assert (
"d2s_train_image_shape" in config["Global"]
), "d2s_train_image_shape must be assigned for static training mode..."
supported_list = ["DB", "SVTR_LCNet", "TableMaster", "LayoutXLM", "SLANet", "SVTR"]
supported_list = [
"DB",
"SVTR_LCNet",
"TableMaster",
"LayoutXLM",
"SLANet",
"SVTR",
"SVTR_HGNet",
"LaTeXOCR",
]
if config["Architecture"]["algorithm"] in ["Distillation"]:
algo = list(config["Architecture"]["Models"].values())[0]["algorithm"]
else: