mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
458f6d684e
* [WIP] Refactor-docs-structure * [Docs] Check link and fix rst (#498) * check link * fix rst * Fix conf.py * Fix lint * minor refine * minor refine * modify zh_readme and index.rst * add migration exampls get_started design to en * add tutorials to en * correct chinese doc link * Apply suggestions from code review * Update docs/en/advanced_tutorials/distributed.md Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
115 lines
1.7 KiB
ReStructuredText
115 lines
1.7 KiB
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
mmengine.model
|
|
===================================
|
|
|
|
.. contents:: mmengine.model
|
|
:depth: 2
|
|
:local:
|
|
:backlinks: top
|
|
|
|
.. currentmodule:: mmengine.model
|
|
|
|
Module
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
BaseModule
|
|
ModuleDict
|
|
ModuleList
|
|
Sequential
|
|
|
|
Model
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
BaseModel
|
|
BaseDataPreprocessor
|
|
ImgDataPreprocessor
|
|
|
|
EMA
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
BaseAveragedModel
|
|
ExponentialMovingAverage
|
|
MomentumAnnealingEMA
|
|
StochasticWeightAverage
|
|
|
|
Model Wrapper
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
MMDistributedDataParallel
|
|
MMSeparateDistributedDataParallel
|
|
MMFullyShardedDataParallel
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
is_model_wrapper
|
|
|
|
Weight Initialization
|
|
----------------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
BaseInit
|
|
Caffe2XavierInit
|
|
ConstantInit
|
|
KaimingInit
|
|
NormalInit
|
|
PretrainedInit
|
|
TruncNormalInit
|
|
UniformInit
|
|
XavierInit
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
bias_init_with_prob
|
|
caffe2_xavier_init
|
|
constant_init
|
|
initialize
|
|
kaiming_init
|
|
normal_init
|
|
trunc_normal_init
|
|
uniform_init
|
|
update_init_info
|
|
xavier_init
|
|
|
|
Utils
|
|
----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
detect_anomalous_params
|
|
merge_dict
|
|
stack_batch
|
|
revert_sync_batchnorm
|