Shortcuts

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.

mmcls.engine

Hooks

ClassNumCheckHook

Class Number Check HOOK.

PreciseBNHook

Precise BN hook.

VisualizationHook

Classification Visualization Hook.

PrepareProtoBeforeValLoopHook

The hook to prepare the prototype in retrievers.

SetAdaptiveMarginsHook

Set adaptive-margins in ArcFaceClsHead based on the power of category-wise count.

Optimizers

Lamb

A pure pytorch variant of FuseLAMB (NvLamb variant) optimizer.

Read the Docs v: 1.x
Versions
master
latest
1.x
dev-1.x
Downloads
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.