mirrors
/
mmselfsup
mirror of https://github.com/open-mmlab/mmselfsup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Fix] Fix import error (#746) 

* fix import error

* update
main
Yixiao Fang 2023-04-10 14:56:52 +08:00 committed by GitHub
parent 6ae4ee5f74
commit c77954bb90
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmselfsup/registry.py
View File

@ -55,15 +55,14 @@ __all__ = [
# Registries For Runner and the related
# manage all kinds of runners like `EpochBasedRunner` and `IterBasedRunner`
RUNNERS = Registry(
'runner', parent=MMENGINE_RUNNERS, locations=['mmselfsup.engine.runner'])
'runner', parent=MMENGINE_RUNNERS, locations=['mmselfsup.engine'])
# manage runner constructors that define how to initialize runners
RUNNER_CONSTRUCTORS = Registry(
'runner constructor',
parent=MMENGINE_RUNNER_CONSTRUCTORS,
locations=['mmselfsup.engine.runner'])
locations=['mmselfsup.engine'])
# manage all kinds of loops like `EpochBasedTrainLoop`
LOOPS = Registry(
'loop', parent=MMENGINE_LOOPS, locations=['mmselfsup.engine.runner'])
LOOPS = Registry('loop', parent=MMENGINE_LOOPS, locations=['mmselfsup.engine'])
# manage all kinds of hooks like `CheckpointHook`
HOOKS = Registry(
'hook', parent=MMENGINE_HOOKS, locations=['mmselfsup.engine.hooks'])