From f86828fe5ac638981d4c492083ea5a1473186734 Mon Sep 17 00:00:00 2001
From: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
Date: Mon, 25 Jul 2022 15:47:53 +0800
Subject: [PATCH] fix custom import error (#371)

---
 configs/benchmarks/classification/_base_/default_runtime.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/configs/benchmarks/classification/_base_/default_runtime.py b/configs/benchmarks/classification/_base_/default_runtime.py
index 7b62ce86..7fb3bd1a 100644
--- 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'),