[Fix] rename custom_hooks_config to custom_hooks (#1427)

* rename custom_hooks_config to custom_hooks
open-mmlab · Jun 15, 2022 · 8890aa3 · 8890aa3
1 parent 07cd4b2
commit 8890aa3
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 5 deletions.
diff --git a/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_224x224_fps30.py b/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_224x224_fps30.py
@@ -17,7 +17,7 @@
 total_epochs = 75
 log_config = dict(interval=10)
 
-custom_hooks_config = [dict(type='ModelSetEpochHook')]
+custom_hooks = [dict(type='ModelSetEpochHook')]
 
 model = dict(
     type='GestureRecognizer',

diff --git a/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_bbox_112x112_fps15.py b/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_bbox_112x112_fps15.py
@@ -17,7 +17,7 @@
 total_epochs = 130
 log_config = dict(interval=10)
 
-custom_hooks_config = [dict(type='ModelSetEpochHook')]
+custom_hooks = [dict(type='ModelSetEpochHook')]
 
 model = dict(
     type='GestureRecognizer',

diff --git a/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_bbox_112x112_fps15_rgb.py b/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_bbox_112x112_fps15_rgb.py
@@ -18,7 +18,7 @@
 total_epochs = 130
 log_config = dict(interval=10)
 
-custom_hooks_config = [dict(type='ModelSetEpochHook')]
+custom_hooks = [dict(type='ModelSetEpochHook')]
 
 model = dict(
     type='GestureRecognizer',

diff --git a/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_bbox_224x224_fps30.py b/configs/hand/gesture_sview_rgbd_vid/mtut/nvgesture/i3d_nvgesture_bbox_224x224_fps30.py
@@ -17,7 +17,7 @@
 total_epochs = 75
 log_config = dict(interval=10)
 
-custom_hooks_config = [dict(type='ModelSetEpochHook')]
+custom_hooks = [dict(type='ModelSetEpochHook')]
 
 model = dict(
     type='GestureRecognizer',

diff --git a/mmpose/apis/train.py b/mmpose/apis/train.py
@@ -168,14 +168,22 @@ def train_model(model,
         else:
             optimizer_config = cfg.optimizer_config
 
+    custom_hooks_cfg = cfg.get('custom_hooks', None)
+    if custom_hooks_cfg is None:
+        custom_hooks_cfg = cfg.get('custom_hooks_config', None)
+        if custom_hooks_cfg is not None:
+            warnings.warn(
+                '"custom_hooks_config" is deprecated, please use '
+                '"custom_hooks" instead.', DeprecationWarning)
+
     # register hooks
     runner.register_training_hooks(
         cfg.lr_config,
         optimizer_config,
         cfg.checkpoint_config,
         cfg.log_config,
         cfg.get('momentum_config', None),
-        custom_hooks_config=cfg.get('custom_hooks_config', None))
+        custom_hooks_config=custom_hooks_cfg)
 
     if distributed:
         runner.register_hook(DistSamplerSeedHook())