提交 2a4708eb 编写于 作者: W wjj19950828

fixed bug for CI

上级 0acebeef
...@@ -61,7 +61,7 @@ def _rename_or_remove_weight(weights, ...@@ -61,7 +61,7 @@ def _rename_or_remove_weight(weights,
Returns: Returns:
None None
''' '''
if origin_name in rename_mapper: if rename_mapper is not None and origin_name in rename_mapper:
origin_name = rename_mapper[origin_name] origin_name = rename_mapper[origin_name]
is_remove = False is_remove = False
if origin_name not in weights: if origin_name not in weights:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册