diff --git a/mmdet/apis/train.py b/mmdet/apis/train.py index 8f2c087415a04437583b3dd61832151b06271754..39982d01ac1f8782a23a3e1c9f5e7ce2219c84af 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 0f4b7416104e32050bd82b2c383448db765772e9..f6d104e33ae64efdab4ea69a8715cce4ac2254c7 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