Skip to content

GitLab

Unverified Commit 68708cfb authored by Harry's avatar Harry Committed by GitHub
Browse files

fix: update module wrapper register_module (#357)

parent 12aab3e1
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/parallel/registry.py
View file @ 68708cfb
......@@ -3,5 +3,5 @@ from torch.nn.parallel import DataParallel, DistributedDataParallel
from mmcv.utils import Registry
MODULE_WRAPPERS = Registry('module wrapper')
MODULE_WRAPPERS.register_module(DataParallel)
MODULE_WRAPPERS.register_module(DistributedDataParallel)
MODULE_WRAPPERS.register_module(module=DataParallel)
MODULE_WRAPPERS.register_module(module=DistributedDataParallel)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment