Shortcuts

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

Read the Docs v: v0.5.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.