Skip to content
Snippets Groups Projects
Unverified Commit c55e53bc authored by David de la Iglesia Castro's avatar David de la Iglesia Castro Committed by GitHub
Browse files

Momentum updater hook (#2571)

* Pass momentum_config to register_training_hooks

* Bump mmcv version

* Explicit default
parent fdc0c6d5
No related branches found
No related tags found
No related merge requests found
......@@ -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())
......
matplotlib
mmcv>=0.4.0
mmcv>=0.4.4
numpy
# need older pillow until torchvision is fixed
Pillow<=6.2.2
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment