fix custom import error (#371)

2022-07-25 15:47:53 +08:00 · 2022-07-25 15:47:53 +08:00 · f86828fe5a
parent 067fbce0b5
commit f86828fe5a
1 changed files with 2 additions and 1 deletions
--- a/configs/benchmarks/classification/_base_/default_runtime.py
+++ b/configs/benchmarks/classification/_base_/default_runtime.py
@ -1,6 +1,7 @@
 default_scope = 'mmcls'
 custom_imports = dict(
-    imports=['mmselfsup.models', 'mmselfsup.core'], allow_failed_imports=False)
+    imports=['mmselfsup.models', 'mmselfsup.engine'],
+    allow_failed_imports=False)

 default_hooks = dict(
    runtime_info=dict(type='RuntimeInfoHook'),