Update __init__.py

sml2h3 · Sep 26, 2022 · 59e236d · 59e236d
1 parent 8f1a204
commit 59e236d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nets/__init__.py b/nets/__init__.py
@@ -215,7 +215,7 @@ def get_random_tensor(self):
     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