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.NETWORK,
'torchvision_' + _cls.__name__):
continue
ClassFactory.register_cls(_cls,
ClassType.NETWORK,
alias='torchvision_' + _cls.__name__)
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))