SyncBuffersHook¶
- class mmengine.hooks.SyncBuffersHook[source]¶
Synchronize model buffers such as running_mean and running_var in BN at the end of each epoch.
- Return type
None
- after_train_epoch(runner)[source]¶
All-reduce model buffers at the end of each epoch.
- Parameters
runner (Runner) – The runner of the training process.
- Return type
None
- before_val_epoch(runner)[source]¶
All-reduce model buffers before each validation epoch.
Synchronize the buffers before each validation if they have not been synchronized at the end of the previous training epoch. This method will be called when using IterBasedTrainLoop.
- Parameters
runner (Runner) – The runner of the training process.
- Return type
None