Shortcuts

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.

PrepareTTAHook

Wraps runner.model with subclass of BaseTTAModel in before_test.

Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.