diff --git a/nets/__init__.py b/nets/__init__.py
index ae55401a..a2761a40 100644
--- 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