diff --git a/ppocr/modeling/architectures/__init__.py b/ppocr/modeling/architectures/__init__.py index 50260f9e4b..cc7e706833 100755 --- a/ppocr/modeling/architectures/__init__.py +++ b/ppocr/modeling/architectures/__init__.py @@ -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: