feat: ui + docs: для жюри

2026-09-23 23:12:10 +03:00
parent 121a281fc6
commit 88a6117eda
7 changed files with 796 additions and 5 deletions
@@ -23,11 +23,12 @@ torch.onnx.export(
    str(OUT / "model.onnx"),
    input_names=["input_ids", "attention_mask", "token_type_ids"],
    output_names=["logits"],
-    dynamic_shapes=[
-        {0: "batch", 1: "seq"},
-        {0: "batch", 1: "seq"},
-        {0: "batch", 1: "seq"},
-    ],
+    dynamic_axes={
+        "input_ids": {0: "batch", 1: "seq"},
+        "attention_mask": {0: "batch", 1: "seq"},
+        "token_type_ids": {0: "batch", 1: "seq"},
+        "logits": {0: "batch", 1: "seq"},
+    },
    opset_version=14,
 )
 print("ONNX exported to", OUT / "model.onnx")