change support list (#14293)
parent
b6bcde143d
commit
8fdc409edf
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue