Shortcuts

EmptyCacheHook

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

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

参数
  • 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.

返回类型

None

Read the Docs v: latest
Versions
latest
stable
Downloads
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.