def register(self, regclass, reg_path, reg_id):
"""Register a component class."""
reg_id = self.get_reg_name(reg_id)
ClassFactory.register_cls(regclass,
type_name=get_reg_type(reg_path),
alias=reg_id)
self.logger.debug('registered: {}'.format(reg_id))
def _register_models_from_current_module_scope(module):
for _name in dir(module):
if _name.startswith("_"):
continue
_cls = getattr(module, _name)
if isinstance(_cls, ModuleType):
continue
if ClassFactory.is_exists(ClassType.SEARCH_SPACE, 'torchvision_' + _cls.__name__):
continue
ClassFactory.register_cls(_cls, ClassType.SEARCH_SPACE, alias='torchvision_' + _cls.__name__)
