Momentum updater hook (open-mmlab#2571)

* Pass momentum_config to register_training_hooks * Bump mmcv version * Explicit default
mike112223 · Apr 30, 2020 · 2eca95e · 2eca95e
1 parent 7d269c1
commit 2eca95e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mmdet/apis/train.py b/mmdet/apis/train.py
@@ -140,7 +140,8 @@ def train_detector(model,
 
     # register hooks
     runner.register_training_hooks(cfg.lr_config, optimizer_config,
-                                   cfg.checkpoint_config, cfg.log_config)
+                                   cfg.checkpoint_config, cfg.log_config,
+                                   cfg.get('momentum_config', None))
     if distributed:
         runner.register_hook(DistSamplerSeedHook())
 

diff --git a/requirements/runtime.txt b/requirements/runtime.txt
@@ -1,5 +1,5 @@
 matplotlib
-mmcv>=0.4.0
+mmcv>=0.4.4
 numpy
 # need older pillow until torchvision is fixed
 Pillow<=6.2.2