[Fix] Fix ut (#422)

* add register all modules

* add scope

tests/test_engine/test_hooks/test_densecl_hook.py

@@ -94,7 +94,8 @@ class TestDenseCLHook(TestCase):
             custom_hooks=[densecl_hook],
             default_hooks=dict(logger=None),
             log_processor=dict(window_size=1),
-            experiment_name='test_densecl_hook')
+            experiment_name='test_densecl_hook',
+            default_scope='mmselfsup')
 
         runner.train()
 

tests/test_engine/test_hooks/test_simsiam_hook.py

@@ -104,7 +104,8 @@ class TestSimSiamHook(TestCase):
             custom_hooks=[simsiam_hook],
             default_hooks=dict(logger=None),
             log_processor=dict(window_size=1),
-            experiment_name='test_simsiam_hook')
+            experiment_name='test_simsiam_hook',
+            default_scope='mmselfsup')
 
         runner.train()
 

tests/test_engine/test_hooks/test_swav_hook.py

@@ -107,7 +107,8 @@ class TestSwAVHook(TestCase):
             custom_hooks=[swav_hook],
             default_hooks=dict(logger=None),
             log_processor=dict(window_size=1),
-            experiment_name='test_swav_hook')
+            experiment_name='test_swav_hook',
+            default_scope='mmselfsup')
 
         runner.train()
 

tests/test_models/test_algorithms/test_barlowtwins.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms.barlowtwins import BarlowTwins
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='ResNet',

tests/test_models/test_algorithms/test_byol.py

@@ -7,7 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms.byol import BYOL
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
 
+register_all_modules()
 backbone = dict(
     type='ResNet',
     depth=18,

tests/test_models/test_algorithms/test_cae.py

@@ -7,6 +7,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms import CAE
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 # model settings
 backbone = dict(type='CAEViT', arch='b', patch_size=16, init_values=0.1)

tests/test_models/test_algorithms/test_deepcluster.py

@@ -8,6 +8,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms import DeepCluster
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 num_classes = 5
 with_sobel = True,

tests/test_models/test_algorithms/test_densecl.py

@@ -9,6 +9,9 @@ import torch
 import mmselfsup
 from mmselfsup.models.algorithms.densecl import DenseCL
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 queue_len = 32
 feat_dim = 2

tests/test_models/test_algorithms/test_mae.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms.mae import MAE
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(type='MAEViT', arch='b', patch_size=16, mask_ratio=0.75)
 neck = dict(

tests/test_models/test_algorithms/test_moco.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms import MoCo
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 queue_len = 32
 feat_dim = 2

tests/test_models/test_algorithms/test_mocov3.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models import MoCoV3
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='MoCoV3ViT',

tests/test_models/test_algorithms/test_npid.py

@@ -8,6 +8,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms import NPID
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='ResNet',

tests/test_models/test_algorithms/test_odc.py

@@ -8,6 +8,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms import ODC
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 num_classes = 5
 backbone = dict(

tests/test_models/test_algorithms/test_relative_loc.py

@@ -7,6 +7,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms.relative_loc import RelativeLoc
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='ResNet',

tests/test_models/test_algorithms/test_rotation_pred.py

@@ -8,6 +8,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms.rotation_pred import RotationPred
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='ResNet',

tests/test_models/test_algorithms/test_simclr.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms.simclr import SimCLR
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='ResNet',

tests/test_models/test_algorithms/test_simmim.py

@@ -7,6 +7,9 @@ from mmengine.data import InstanceData
 
 from mmselfsup.models.algorithms import SimMIM
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 
 @pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')

tests/test_models/test_algorithms/test_simsiam.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms import SimSiam
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 backbone = dict(
     type='ResNet',

tests/test_models/test_algorithms/test_swav.py

@@ -7,6 +7,9 @@ import torch
 
 from mmselfsup.models.algorithms.swav import SwAV
 from mmselfsup.structures import SelfSupDataSample
+from mmselfsup.utils import register_all_modules
+
+register_all_modules()
 
 nmb_crops = [2, 6]
 backbone = dict(