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.

Read the Docs v: v0.3.0
Versions
latest
stable
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.