From 9c395ac802d0bbca9e24bbe9db780e48029514dc Mon Sep 17 00:00:00 2001 From: Bin Lu Date: Tue, 1 Jun 2021 09:52:56 +0800 Subject: [PATCH] Update __init__.py --- ppcls/arch/neck/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/arch/neck/__init__.py b/ppcls/arch/neck/__init__.py index 7195c9dc..93529b72 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 -- GitLab