Skip to content

Commit

Permalink
change support list (#14293)
Browse files Browse the repository at this point in the history
  • Loading branch information
liuhongen1234567 authored Nov 29, 2024
1 parent b6bcde1 commit 8fdc409
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion ppocr/modeling/architectures/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 8fdc409

Please sign in to comment.