Replace torch.onnx._export call with torch.onnx.export

Deprecated torch.onnx._export
ModECI · Jan 15, 2025 · 63ee3cf · 63ee3cf
1 parent c650f5b
commit 63ee3cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/MDF/abcd_torch.py b/examples/MDF/abcd_torch.py
@@ -86,7 +86,7 @@ def forward(self, input: torch.Tensor):
 
 # Export the model
 fn = "ABCD_from_torch.onnx"
-torch_out = torch.onnx._export(
+torch.onnx.export(
     m_abcd,  # model being run
     input,  # model input (or a tuple for multiple inputs)
     fn,  # where to save the model (can be a file or file-like object)