Text2Video-Zero

Runtime error

App Files Files Community

Text2Video-Zero / annotator /uniformer /mmcv /parallel /registry.py

shadow

lev1

Duplicate from PAIR/Text2Video-Zero

d0a405c over 3 years ago

Raw History Blame Contribute Delete

332 Bytes

# Copyright (c) OpenMMLab. All rights reserved.
from torch.nn.parallel import DataParallel, DistributedDataParallel

from annotator.uniformer.mmcv.utils import Registry

MODULE_WRAPPERS = Registry('module wrapper')
MODULE_WRAPPERS.register_module(module=DataParallel)
MODULE_WRAPPERS.register_module(module=DistributedDataParallel)

	# Copyright (c) OpenMMLab. All rights reserved.
	from torch.nn.parallel import DataParallel, DistributedDataParallel

	from annotator.uniformer.mmcv.utils import Registry

	MODULE_WRAPPERS = Registry('module wrapper')
	MODULE_WRAPPERS.register_module(module=DataParallel)
	MODULE_WRAPPERS.register_module(module=DistributedDataParallel)