Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
engine.rst 947 Bytes
Newer Older
limm's avatar
add resources part  
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51 
.. 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
   LARS
   LearningRateDecayOptimWrapperConstructor