未验证 提交 9c395ac8 编写于 作者: B Bin Lu 提交者: GitHub

Update __init__.py

上级 271f7946
......@@ -17,7 +17,7 @@ __all__ = ['build_neck"]
def build_neck(config):
support_dict = ['FPN', 'FC']
module_name = config.pop('name')
assert module_name in support_dict, Exception('head only support {}'.format(
assert module_name in support_dict, Exception('neck only support {}'.format(
support_dict))
module_class = eval(module_name)(**config)
return module_class
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册