mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
08dc8c75d3
* remove basehead * add moco series * add byol simclr simsiam * add ut * update configs * add simsiam hook * add and refactor beit * update ut * add cae * update extract_feat * refactor cae * add mae * refactor data preprocessor * update heads * add maskfeat * add milan * add simmim * add mixmim * fix lint * fix ut * fix lint * add eva * add densecl * add barlowtwins * add swav * fix lint * update readtherdocs rst * update docs * update * Decrease UT memory usage * Fix docstring * update DALLEEncoder * Update model docs * refactor dalle encoder * update docstring * fix ut * fix config error * add val_cfg and test_cfg * refactor clip generator * fix lint * pass check * fix ut * add lars * update type of BEiT in configs * Use MMEngine style momentum in EMA. * apply mmpretrain solarize --------- Co-authored-by: mzr1996 <mzr1996@163.com>
50 lines
895 B
ReStructuredText
50 lines
895 B
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
.. module:: mmpretrain.engine
|
|
|
|
mmpretrain.engine
|
|
===================================
|
|
|
|
This package includes some runtime components, including hooks, runners, optimizers and loops. These components are useful in
|
|
classification tasks but not supported by MMEngine yet.
|
|
|
|
.. note::
|
|
|
|
Some components may be moved to MMEngine in the future.
|
|
|
|
.. contents:: mmpretrain.engine
|
|
:depth: 2
|
|
:local:
|
|
:backlinks: top
|
|
|
|
.. module:: mmpretrain.engine.hooks
|
|
|
|
Hooks
|
|
------------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
ClassNumCheckHook
|
|
PreciseBNHook
|
|
VisualizationHook
|
|
PrepareProtoBeforeValLoopHook
|
|
SetAdaptiveMarginsHook
|
|
EMAHook
|
|
SimSiamHook
|
|
DenseCLHook
|
|
SwAVHook
|
|
|
|
.. module:: mmpretrain.engine.optimizers
|
|
|
|
Optimizers
|
|
------------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
Lamb
|