Попытка добавить onnx в работе с nvidia

2026-04-15 17:53:06 +05:00
parent 0c871c2314
commit 7addcf051c
13 changed files with 2484 additions and 371 deletions
--- a/onnx_export.py
+++ b/onnx_export.py
@@ -0,0 +1,8 @@
+import torch
+from src.export_to_onnx import export_to_onnx
+from src.config.presets import SMALL
+
+
+if __name__ == "__main__":
+    device = torch.device("cuda")
+    export_to_onnx(SMALL, "src/pretrained/amt_s.onnx", device)