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

before_val_epoch(runner)[源代码]

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.

参数

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

返回类型

None

Read the Docs v: stable
Versions
latest
stable
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.