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.

备注

Some components may be moved to MMEngine in the future.

Hooks

ClassNumCheckHook	Class Number Check HOOK.
PreciseBNHook	Precise BN hook.
VisualizationHook	Classification Visualization Hook.

Optimizers

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