mmengine.hooks

Hook	Base hook class.
CheckpointHook	Save checkpoints periodically.
EMAHook	A Hook to apply Exponential Moving Average (EMA) on the model during training.
LoggerHook	Collect logs from different components of Runner and write them to terminal, JSON file, tensorboard and wandb .etc.
NaiveVisualizationHook	Show or Write the predicted results during the process of testing.
ParamSchedulerHook	A hook to update some hyper-parameters in optimizer, e.g., learning rate and momentum.
RuntimeInfoHook	A hook that updates runtime information into message hub.
DistSamplerSeedHook	Data-loading sampler for distributed training.
IterTimerHook	A hook that logs the time spent during iteration.
SyncBuffersHook	Synchronize model buffers such as running_mean and running_var in BN at the end of each epoch.
EmptyCacheHook	Releases all unoccupied cached GPU memory during the process of training.
ProfilerHook	A hook to analyze performance during training and inference.
NPUProfilerHook	NPUProfiler to analyze performance during training.
PrepareTTAHook	Wraps runner.model with subclass of BaseTTAModel in before_test.
EarlyStoppingHook	Early stop the training when the monitored metric reached a plateau.