Update __init__.py
parent
271f79466a
commit
9c395ac802
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue