Shortcuts

EmptyCacheHook

class mmengine.hooks.EmptyCacheHook(before_epoch=False, after_epoch=True, after_iter=False)[source]

Releases all unoccupied cached GPU memory during the process of training.

Parameters
  • before_epoch (bool) – Whether to release cache before an epoch. Defaults to False.

  • after_epoch (bool) – Whether to release cache after an epoch. Defaults to True.

  • after_iter (bool) – Whether to release cache after an iteration. Defaults to False.

Return type

None

Read the Docs v: v0.7.0
Versions
latest
stable
v0.7.0
v0.6.0
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.