提交 89b96861 编写于 作者: C Channingss

add _rename_or_remove_weight

上级 9fc66d34
...@@ -42,7 +42,7 @@ def _const_weight_or_none(node, necessary=False): ...@@ -42,7 +42,7 @@ def _const_weight_or_none(node, necessary=False):
return None 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: if origin_name not in weights:
raise KeyError('{} not a key in {}'.format(origin_name, weights)) raise KeyError('{} not a key in {}'.format(origin_name, weights))
if target_name is None: if target_name is None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册