Shortcuts

Source code for mmpretrain.utils.setup_env

# Copyright (c) OpenMMLab. All rights reserved.
import datetime
import warnings

from mmengine import DefaultScope


[docs]def register_all_modules(init_default_scope: bool = True) -> None: """Register all modules in mmpretrain into the registries. Args: init_default_scope (bool): Whether initialize the mmpretrain default scope. If True, the global default scope will be set to `mmpretrain`, and all registries will build modules from mmpretrain's registry node. To understand more about the registry, please refer to https://github.com/open-mmlab/mmengine/blob/main/docs/en/tutorials/registry.md Defaults to True. """ # noqa: E501 import mmpretrain.datasets # noqa: F401,F403 import mmpretrain.engine # noqa: F401,F403 import mmpretrain.evaluation # noqa: F401,F403 import mmpretrain.models # noqa: F401,F403 import mmpretrain.structures # noqa: F401,F403 import mmpretrain.visualization # noqa: F401,F403 if not init_default_scope: return current_scope = DefaultScope.get_current_instance() if current_scope is None: DefaultScope.get_instance('mmpretrain', scope_name='mmpretrain') elif current_scope.scope_name != 'mmpretrain': warnings.warn( f'The current default scope "{current_scope.scope_name}" ' 'is not "mmpretrain", `register_all_modules` will force ' 'the current default scope to be "mmpretrain". If this is ' 'not expected, please set `init_default_scope=False`.') # avoid name conflict new_instance_name = f'mmpretrain-{datetime.datetime.now()}' DefaultScope.get_instance(new_instance_name, scope_name='mmpretrain')
Read the Docs v: dev
Versions
latest
stable
mmcls-1.x
mmcls-0.x
dev
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.