mmpretrain.engine¶
该包中包含了一些运行时组件,如钩子(hook)、执行器(runner)、优化器(optimizer)和循环执行器(loop)。这些组件在分类任务中需要用到,而还未被 MMEngine 支持。
备注
部分组件未来可能会被移动到 MMEngine 中。
Hooks¶
Class Number Check HOOK. |
|
Precise BN hook. |
|
Classification Visualization Hook. |
|
The hook to prepare the prototype in retrievers. |
|
Set adaptive-margins in ArcFaceClsHead based on the power of category-wise count. |
|
A Hook to apply Exponential Moving Average (EMA) on the model during training. |
|
Hook for SimSiam. |
|
Hook for DenseCL. |
|
Hook for SwAV. |
Optimizers¶
A pure pytorch variant of FuseLAMB (NvLamb variant) optimizer. |
|
Implements layer-wise adaptive rate scaling for SGD. |
|
Different learning rates are set for different layers of backbone. |