[Fix] complete training hooks. (open-mmlab#1236)

* Fix: complete training hooks. * not expose timer_hook Co-authored-by: ly015 <liyining0712@gmail.com>
wjkim81 · Mar 14, 2022 · 663cfc8 · 663cfc8
1 parent 2f9829e
commit 663cfc8
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/mmpose/apis/train.py b/mmpose/apis/train.py
@@ -169,9 +169,14 @@ def train_model(model,
             optimizer_config = cfg.optimizer_config
 
     # register hooks
-    runner.register_training_hooks(cfg.lr_config, optimizer_config,
-                                   cfg.checkpoint_config, cfg.log_config,
-                                   cfg.get('momentum_config', None))
+    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))
+
     if distributed:
         runner.register_hook(DistSamplerSeedHook())