mmpretrain/docs/en/api/engine.rst
Ezra-Yu b0007812d6
[Enhance] Enhance ArcFaceClsHead. (#1181)
* update arcface

* fix unit tests

* add adv-margins

add adv-margins

update arcface

* rebase

* update doc and fix ut

* rebase

* update code

* rebase

* use label data

* update set-margins

* Modify Arcface related method names.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 18:10:39 +08:00

48 lines
881 B
ReStructuredText

.. role:: hidden
:class: hidden-section
.. module:: mmcls.engine
mmcls.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:: mmcls.engine
:depth: 2
:local:
:backlinks: top
.. module:: mmcls.engine.hooks
Hooks
------------------
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
ClassNumCheckHook
PreciseBNHook
VisualizationHook
PrepareProtoBeforeValLoopHook
SetAdaptiveMarginsHook
.. module:: mmcls.engine.optimizers
Optimizers
------------------
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
Lamb