Update __init__.py

2026-08-12 21:21:44 +08:00 · 2022-09-26 15:26:52 +08:00
parent 8f1a204eb8
commit 59e236d503
1 changed files with 1 additions and 1 deletions
@@ -215,7 +215,7 @@ class Net(torch.nn.Module):
    def export_onnx(self, net, dummy_input, graph_path, input_names, output_names, dynamic_ax):
        torch.onnx.export(net, dummy_input, graph_path, export_params=True, verbose=False,
                          input_names=input_names, output_names=output_names, dynamic_axes=dynamic_ax,
-                          opset_version=12, do_constant_folding=True, _retain_param_name=False)
+                          opset_version=12, do_constant_folding=True)


    @staticmethod