We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d872c71 commit 58c7d71Copy full SHA for 58c7d71
tools/export_onnx.py
@@ -88,7 +88,7 @@ def main():
88
logger.info("loading checkpoint done.")
89
dummy_input = torch.randn(args.batch_size, 3, exp.test_size[0], exp.test_size[1])
90
91
- torch.onnx._export(
+ torch.onnx.export(
92
model,
93
dummy_input,
94
args.output_name,
0 commit comments