From 89b96861b95e62bc5479b42300997fc9938f555b Mon Sep 17 00:00:00 2001 From: Channingss Date: Thu, 14 Jan 2021 09:33:49 +0000 Subject: [PATCH] add _rename_or_remove_weight --- x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py index 5d7c516..9a9fe0b 100644 --- a/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/dygraph/onnx2paddle/opset9/opset.py @@ -42,7 +42,7 @@ def _const_weight_or_none(node, necessary=False): return None -def rename_or_remove_weight(weights, origin_name, target_name=None): +def _rename_or_remove_weight(weights, origin_name, target_name=None): if origin_name not in weights: raise KeyError('{} not a key in {}'.format(origin_name, weights)) if target_name is None: -- GitLab