remove deprecate register_module (#262)

mmcv/runner/hooks/checkpoint.py

@@ -5,7 +5,7 @@ from ..dist_utils import master_only
 from .hook import HOOKS, Hook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class CheckpointHook(Hook):
 
     def __init__(self,

mmcv/runner/hooks/closure.py

@@ -2,7 +2,7 @@
 from .hook import HOOKS, Hook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class ClosureHook(Hook):
 
     def __init__(self, fn_name, fn):

mmcv/runner/hooks/iter_timer.py

@@ -4,7 +4,7 @@ import time
 from .hook import HOOKS, Hook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class IterTimerHook(Hook):
 
     def before_epoch(self, runner):

mmcv/runner/hooks/logger/mlflow.py

@@ -6,7 +6,7 @@ from ..hook import HOOKS
 from .base import LoggerHook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class MlflowLoggerHook(LoggerHook):
 
     def __init__(self,

mmcv/runner/hooks/logger/pavi.py

@@ -31,7 +31,7 @@ def is_scalar(val, include_np=True, include_torch=True):
         return False
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class PaviLoggerHook(LoggerHook):
 
     def __init__(self,

mmcv/runner/hooks/logger/tensorboard.py

@@ -8,7 +8,7 @@ from ..hook import HOOKS
 from .base import LoggerHook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class TensorboardLoggerHook(LoggerHook):
 
     def __init__(self,

mmcv/runner/hooks/logger/text.py

@@ -11,7 +11,7 @@ from ..hook import HOOKS
 from .base import LoggerHook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class TextLoggerHook(LoggerHook):
     """Logger hook in text.
 

mmcv/runner/hooks/logger/wandb.py

@@ -6,7 +6,7 @@ from ..hook import HOOKS
 from .base import LoggerHook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class WandbLoggerHook(LoggerHook):
 
     def __init__(self,

mmcv/runner/hooks/lr_updater.py

@@ -114,7 +114,7 @@ class LrUpdaterHook(Hook):
                 self._set_lr(runner, warmup_lr)
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class FixedLrUpdaterHook(LrUpdaterHook):
 
     def __init__(self, **kwargs):
@@ -124,7 +124,7 @@ class FixedLrUpdaterHook(LrUpdaterHook):
         return base_lr
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class StepLrUpdaterHook(LrUpdaterHook):
 
     def __init__(self, step, gamma=0.1, **kwargs):
@@ -154,7 +154,7 @@ class StepLrUpdaterHook(LrUpdaterHook):
         return base_lr * self.gamma**exp
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class ExpLrUpdaterHook(LrUpdaterHook):
 
     def __init__(self, gamma, **kwargs):
@@ -166,7 +166,7 @@ class ExpLrUpdaterHook(LrUpdaterHook):
         return base_lr * self.gamma**progress
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class PolyLrUpdaterHook(LrUpdaterHook):
 
     def __init__(self, power=1., min_lr=0., **kwargs):
@@ -185,7 +185,7 @@ class PolyLrUpdaterHook(LrUpdaterHook):
         return (base_lr - self.min_lr) * coeff + self.min_lr
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class InvLrUpdaterHook(LrUpdaterHook):
 
     def __init__(self, gamma, power=1., **kwargs):
@@ -198,7 +198,7 @@ class InvLrUpdaterHook(LrUpdaterHook):
         return base_lr * (1 + self.gamma * progress)**(-self.power)
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class CosineAnealingLrUpdaterHook(LrUpdaterHook):
 
     def __init__(self, min_lr=None, min_lr_ratio=None, **kwargs):
@@ -221,7 +221,7 @@ class CosineAnealingLrUpdaterHook(LrUpdaterHook):
         return annealing_cos(base_lr, target_lr, progress / max_progress)
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class CyclicLrUpdaterHook(LrUpdaterHook):
     """Cyclic LR Scheduler
 

mmcv/runner/hooks/memory.py

@@ -4,7 +4,7 @@ import torch
 from .hook import HOOKS, Hook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class EmptyCacheHook(Hook):
 
     def __init__(self, before_epoch=False, after_epoch=True, after_iter=False):

mmcv/runner/hooks/momentum_updater.py

@@ -103,7 +103,7 @@ class MomentumUpdaterHook(Hook):
                 self._set_momentum(runner, warmup_momentum)
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class CosineAnealingMomentumUpdaterHook(MomentumUpdaterHook):
 
     def __init__(self, min_momentum=None, min_momentum_ratio=None, **kwargs):
@@ -127,7 +127,7 @@ class CosineAnealingMomentumUpdaterHook(MomentumUpdaterHook):
                              progress / max_progress)
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class CyclicMomentumUpdaterHook(MomentumUpdaterHook):
     """Cyclic momentum Scheduler
 

mmcv/runner/hooks/optimizer.py

@@ -4,7 +4,7 @@ from torch.nn.utils import clip_grad
 from .hook import HOOKS, Hook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class OptimizerHook(Hook):
 
     def __init__(self, grad_clip=None):

mmcv/runner/hooks/sampler_seed.py

@@ -2,7 +2,7 @@
 from .hook import HOOKS, Hook
 
 
-@HOOKS.register_module
+@HOOKS.register_module()
 class DistSamplerSeedHook(Hook):
 
     def before_epoch(self, runner):

tests/test_registry.py

@@ -9,7 +9,7 @@ def test_registry():
     assert CATS.module_dict == {}
     assert len(CATS) == 0
 
-    @CATS.register_module
+    @CATS.register_module()
     class BritishShorthair:
         pass
 
@@ -124,14 +124,14 @@ def test_registry():
 def test_build_from_cfg():
     BACKBONES = mmcv.Registry('backbone')
 
-    @BACKBONES.register_module
+    @BACKBONES.register_module()
     class ResNet:
 
         def __init__(self, depth, stages=4):
             self.depth = depth
             self.stages = stages
 
-    @BACKBONES.register_module
+    @BACKBONES.register_module()
     class ResNeXt:
 
         def __init__(self, depth, stages=4):