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.

Read the Docs v: v0.10.4
Versions
latest
stable
v0.10.4
v0.10.3
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
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.