Support variable length input

2026-09-24 23:33:08 +08:00 · 2022-08-15 09:00:35 +00:00
parent ebe3e885c0
commit 7efee821bc
1 changed files with 2 additions and 0 deletions
@@ -140,6 +140,8 @@ def export_single_model(model,
            infer_shape = [3, 488, 488]
            if arch_config["algorithm"] == "TableMaster":
                infer_shape = [3, 480, 480]
+            if arch_config["algorithm"] == "SLANet":
+                infer_shape = [3, -1, -1]
        model = to_static(
            model,
            input_spec=[