Skip to content

Commit 58c7d71

Browse files
committed
Rename method name
1 parent d872c71 commit 58c7d71

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/export_onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def main():
8888
logger.info("loading checkpoint done.")
8989
dummy_input = torch.randn(args.batch_size, 3, exp.test_size[0], exp.test_size[1])
9090

91-
torch.onnx._export(
91+
torch.onnx.export(
9292
model,
9393
dummy_input,
9494
args.output_name,

0 commit comments

Comments
 (0)