Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmdetection3d
Commits
b16c8dfa
Unverified
Commit
b16c8dfa
authored
Aug 09, 2022
by
ChaimZhu
Committed by
GitHub
Aug 09, 2022
Browse files
add log processor in registry (#1709)
parent
f6ef8b73
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
0 deletions
+10
-0
mmdet3d/registry.py
mmdet3d/registry.py
+10
-0
No files found.
mmdet3d/registry.py
View file @
b16c8dfa
...
...
@@ -8,13 +8,16 @@ https://mmengine.readthedocs.io/en/latest/tutorials/registry.html.
from
mmengine.registry
import
DATA_SAMPLERS
as
MMENGINE_DATA_SAMPLERS
from
mmengine.registry
import
DATASETS
as
MMENGINE_DATASETS
from
mmengine.registry
import
EVALUATOR
as
MMENGINE_EVALUATOR
from
mmengine.registry
import
HOOKS
as
MMENGINE_HOOKS
from
mmengine.registry
import
LOG_PROCESSORS
as
MMENGINE_LOG_PROCESSORS
from
mmengine.registry
import
LOOPS
as
MMENGINE_LOOPS
from
mmengine.registry
import
METRICS
as
MMENGINE_METRICS
from
mmengine.registry
import
MODEL_WRAPPERS
as
MMENGINE_MODEL_WRAPPERS
from
mmengine.registry
import
MODELS
as
MMENGINE_MODELS
from
mmengine.registry
import
\
OPTIM_WRAPPER_CONSTRUCTORS
as
MMENGINE_OPTIM_WRAPPER_CONSTRUCTORS
from
mmengine.registry
import
OPTIM_WRAPPERS
as
MMENGINE_OPTIM_WRAPPERS
from
mmengine.registry
import
OPTIMIZERS
as
MMENGINE_OPTIMIZERS
from
mmengine.registry
import
PARAM_SCHEDULERS
as
MMENGINE_PARAM_SCHEDULERS
from
mmengine.registry
import
\
...
...
@@ -53,6 +56,8 @@ WEIGHT_INITIALIZERS = Registry(
# mangage all kinds of optimizers like `SGD` and `Adam`
OPTIMIZERS
=
Registry
(
'optimizer'
,
parent
=
MMENGINE_OPTIMIZERS
)
# manage optimizer wrapper
OPTIM_WRAPPERS
=
Registry
(
'optim wrapper'
,
parent
=
MMENGINE_OPTIM_WRAPPERS
)
# manage constructors that customize the optimization hyperparameters.
OPTIM_WRAPPER_CONSTRUCTORS
=
Registry
(
'optimizer wrapper constructor'
,
...
...
@@ -62,6 +67,8 @@ PARAM_SCHEDULERS = Registry(
'parameter scheduler'
,
parent
=
MMENGINE_PARAM_SCHEDULERS
)
# manage all kinds of metrics
METRICS
=
Registry
(
'metric'
,
parent
=
MMENGINE_METRICS
)
# manage evaluator
EVALUATOR
=
Registry
(
'evaluator'
,
parent
=
MMENGINE_EVALUATOR
)
# manage task-specific modules like anchor generators and box coders
TASK_UTILS
=
Registry
(
'task util'
,
parent
=
MMENGINE_TASK_UTILS
)
...
...
@@ -70,3 +77,6 @@ TASK_UTILS = Registry('task util', parent=MMENGINE_TASK_UTILS)
VISUALIZERS
=
Registry
(
'visualizer'
,
parent
=
MMENGINE_VISUALIZERS
)
# manage visualizer backend
VISBACKENDS
=
Registry
(
'vis_backend'
,
parent
=
MMENGINE_VISBACKENDS
)
# manage logprocessor
LOG_PROCESSORS
=
Registry
(
'log_processor'
,
parent
=
MMENGINE_LOG_PROCESSORS
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment