提交 7efee821 编写于 作者: 文幕地方's avatar 文幕地方

Support variable length input

上级 ebe3e885
...@@ -140,6 +140,8 @@ def export_single_model(model, ...@@ -140,6 +140,8 @@ def export_single_model(model,
infer_shape = [3, 488, 488] infer_shape = [3, 488, 488]
if arch_config["algorithm"] == "TableMaster": if arch_config["algorithm"] == "TableMaster":
infer_shape = [3, 480, 480] infer_shape = [3, 480, 480]
if arch_config["algorithm"] == "SLANet":
infer_shape = [3, -1, -1]
model = to_static( model = to_static(
model, model,
input_spec=[ input_spec=[
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册