From 92847d0b8180f0ab89cc0b14236b94c139c9c1ad Mon Sep 17 00:00:00 2001 From: Channingss Date: Tue, 19 Jan 2021 06:23:53 +0000 Subject: [PATCH] add comment of function:_rename_or_remove_weight --- x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py index 804c9b4..b3f7bca 100644 --- a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py @@ -48,8 +48,10 @@ def _rename_or_remove_weight(weights, origin_name, target_name=None, is_remove=T Args: weights(dict[String:np.ndarray]): Dict stored paramters, the key in weights is name of parameter. - origin_name(String): Name of parameter to eename or remove. - target_name(String, optional): if target_name is not None, add new key-value pair {target_name:weights[origin_name]} to weights, and target_name must follow paddle's naming rule of parameters. Default: None. + origin_name(String): Name of parameter to rename or remove. + target_name(String, optional): if target_name is not None, add new key-value pair + {target_name:weights[origin_name]} to weights, and target_name must follow paddle's + naming rule of parameters. Default: None. is_remove: if is_remove is True, remove origin key-value pair. Default: True. Returns: None -- GitLab