Rename method name

Megvii-BaseDetection · Nov 27, 2024 · 58c7d71 · 58c7d71
1 parent d872c71
commit 58c7d71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/export_onnx.py b/tools/export_onnx.py
@@ -88,7 +88,7 @@ def main():
     logger.info("loading checkpoint done.")
     dummy_input = torch.randn(args.batch_size, 3, exp.test_size[0], exp.test_size[1])
 
-    torch.onnx._export(
+    torch.onnx.export(
         model,
         dummy_input,
         args.output_name,