Shortcuts

SyncBuffersHook

class mmengine.hooks.SyncBuffersHook[源代码]

Synchronize model buffers such as running_mean and running_var in BN at the end of each epoch.

返回类型

None

after_train_epoch(runner)[源代码]

All-reduce model buffers at the end of each epoch.

参数

runner (Runner) – The runner of the training process.

返回类型

None

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.