mmpretrain.engine

该包中包含了一些运行时组件，如钩子（hook）、执行器（runner）、优化器（optimizer）和循环执行器（loop）。这些组件在分类任务中需要用到，而还未被 MMEngine 支持。

备注

部分组件未来可能会被移动到 MMEngine 中。

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.
EMAHook	A Hook to apply Exponential Moving Average (EMA) on the model during training.
SimSiamHook	Hook for SimSiam.
DenseCLHook	Hook for DenseCL.
SwAVHook	Hook for SwAV.

Optimizers

Lamb	A pure pytorch variant of FuseLAMB (NvLamb variant) optimizer.
LARS	Implements layer-wise adaptive rate scaling for SGD.
LearningRateDecayOptimWrapperConstructor	Different learning rates are set for different layers of backbone.