diff --git a/ppcls/arch/neck/__init__.py b/ppcls/arch/neck/__init__.py index 7195c9dcd..93529b727 100644 --- a/ppcls/arch/neck/__init__.py +++ b/ppcls/arch/neck/__init__.py @@ -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