| 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. |