From c55e53bc64c9eb0e3a55637f4db2965714a5d0ea Mon Sep 17 00:00:00 2001
From: David de la Iglesia Castro <daviddelaiglesiacastro@gmail.com>
Date: Thu, 30 Apr 2020 19:35:48 +0200
Subject: [PATCH] Momentum updater hook (#2571)
* Pass momentum_config to register_training_hooks
* Bump mmcv version
* Explicit default
---
mmdet/apis/train.py | 3 ++-
requirements/runtime.txt | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/mmdet/apis/train.py b/mmdet/apis/train.py
index 8f2c0874..39982d01 100644
--- 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
index 0f4b7416..f6d104e3 100644
--- 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
--
GitLab